Update Dockerfile

Dockerfile

@@ -2,9 +2,18 @@
 FROM python:3.9.13
 
 # Install system dependencies
-RUN apt-get update && \
-    apt-get install -y \
+# RUN apt-get update && \
+#     apt-get install -y \
+#     ffmpeg \
+#     libgl1 \
+#     && apt-get clean \
+#     && rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && apt-get install -y \
     ffmpeg \
+    git \
+    curl \
+    wget \
     libgl1 \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
@@ -13,10 +22,16 @@ RUN apt-get update && \
 RUN useradd -m -u 1000 user
 USER user
 
-# Set environment variables
+# # Set environment variables
+# ENV HOME=/home/user \
+#     PATH=/home/user/.local/bin:$PATH \
+#     NUMBA_CACHE_DIR=/tmp/numba_cache
+
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
-    NUMBA_CACHE_DIR=/tmp/numba_cache
+    NUMBA_CACHE_DIR=/tmp/numba_cache \
+    NVIDIA_VISIBLE_DEVICES=all \
+    NVIDIA_DRIVER_CAPABILITIES=compute,utility
 
 # Set working directory
 WORKDIR $HOME/app
@@ -35,4 +50,5 @@ RUN pip install huggingface_hub && \
 EXPOSE 7860
 
 # Start the application with Gunicorn
-CMD ["gunicorn", "-b", "0.0.0.0:7860", "--timeout", "300", "app:app"]
+# CMD ["gunicorn", "-b", "0.0.0.0:7860", "--timeout", "300", "app:app"]
+CMD ["gunicorn", "-w", "2", "-k", "gevent", "-b", "0.0.0.0:7860", "--timeout", "600", "app:app"]