Upload Dockerfile

Dockerfile

@@ -1,17 +1,20 @@
-# Use a base image with Python and CUDA for GPU support (recommended for Qwen 2.5 3B)
-# This image includes Python, CUDA, and common ML libraries
-FROM nvcr.io/nvidia/pytorch:24.01-py3
+# Use a slim Python base image for fast builds and small footprint
+FROM python:3.11-slim
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
-# Hugging Face Spaces uses port 7860 by default for web applications
+# Hugging Face Spaces uses port 7860 by default
 ENV PORT=7860
 
 # Set working directory
 WORKDIR /app
 
+# Install system dependencies if needed (minimal for this app)
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
 # Copy requirements and install Python dependencies
-# We use --no-cache-dir to keep the image size small
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
@@ -21,6 +24,5 @@ COPY . .
 # Expose the port
 EXPOSE 7860
 
-# Command to run the application (matches the Procfile logic)
-# We use the explicit port 7860 as required by HF Spaces
+# Command to run the application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]