fixed package issue in dockerfile

.huggingface.yml

@@ -0,0 +1,2 @@
+build:
+  docker: true

Dockerfile

@@ -1,9 +1,7 @@
 FROM python:3.10-slim
 
-# Set working directory
 WORKDIR /app
 
-# Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     git-lfs \
@@ -16,25 +14,15 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/* \
     && git lfs install
 
-# Copy requirements first for better caching
 COPY requirements.txt .
-
-# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application files
 COPY . .
 
-# Create necessary directories
 RUN mkdir -p checkpoints uploads
 
-# Verify model files exist and show their sizes
-RUN ls -la checkpoints/ || echo "Checkpoints directory not found"
-RUN test -f checkpoints/efficientnet.onnx && echo "efficientnet.onnx found" || echo "efficientnet.onnx missing"
-RUN test -f checkpoints/model.pth && echo "model.pth found" || echo "model.pth missing"
-
-# Expose the port Flask runs on
-EXPOSE 5000
+RUN ls -la checkpoints/ || echo "Checkpoints directory not found" && \
+    test -f checkpoints/efficientnet.onnx && echo "efficientnet.onnx found" || echo "efficientnet.onnx missing" && \
+    test -f checkpoints/model.pth && echo "model.pth found" || echo "model.pth missing"
 
-# Run the application
-CMD ["python", "app.py"]
+# No EXPOSE or CMD — Hugging Face runs `python app.py` for Gradio automatically

requirements.txt

@@ -9,4 +9,5 @@ albumentations
 opencv-python
 torchsummary
 onnx
-onnx2pytorch
+onnx2pytorch
+gradio