Update Dockerfile

Dockerfile

@@ -1,11 +1,18 @@
-FROM python:3.10
+# Use a Python base image with CUDA support for GPU
+FROM python:3.10-slim
 
+# Set working directory
 WORKDIR /app
 
+# Copy requirements and install dependencies
 COPY requirements.txt .
-RUN pip install -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
-COPY . .
+# Copy the FastAPI app
+COPY main.py .
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Expose port 8000 (default for Hugging Face Spaces)
+EXPOSE 8000
 
+# Command to run the app with gunicorn
+CMD ["gunicorn", "-w", "4", "-k", "uvicorn.workers.UvicornWorker", "main:app", "--bind", "0.0.0.0:8000"]