Update Dockerfile

Dockerfile

@@ -1,13 +1,39 @@
-FROM python:3.10
+FROM nvidia/cuda:12.1-runtime-ubuntu22.04
 
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV DEBIAN_FRONTEND=noninteractive
+ENV HF_HOME=/app/cache
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    python3 \
+    python3-pip \
+    python3-venv \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
 WORKDIR /app
 
+# Copy requirements and install Python dependencies
 COPY requirements.txt .
 
-RUN pip install --no-cache-dir -r requirements.txt
+# Install Python dependencies
+RUN pip3 install --no-cache-dir -r requirements.txt
 
+# Copy application code
 COPY . .
 
-EXPOSE 7860
+# Create cache directory
+RUN mkdir -p /app/cache
+
+# Expose ports
+EXPOSE 7860 5000
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:5000/health || exit 1
 
-CMD ["python", "app.py"]
+# Run both Gradio and Flask servers
+CMD ["python3", "app.py"]