Update Dockerfile

Dockerfile

@@ -1,14 +1,33 @@
 FROM python:3.10-slim
 
+# Set working directory
 WORKDIR /app
 
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    g++ \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements first for better caching
 COPY requirements.txt .
-RUN apt-get update && apt-get install -y git && \
-    pip install --no-cache-dir -r requirements.txt
 
-COPY main.py .
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
+COPY . .
 
+# Create cache directory for models
+RUN mkdir -p /root/.cache/huggingface/hub
+
+# Expose port
 EXPOSE 7860
-EXPOSE 8000
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+# Run the application
+CMD ["python", "-u", "app.py"]