Update Dockerfile

Dockerfile

@@ -1,3 +1,34 @@
+# FROM nvidia/cuda:12.1.1-runtime-ubuntu22.04
+# ENV DEBIAN_FRONTEND=noninteractive
+
+# # Install system dependencies
+# RUN apt-get update && apt-get install -y \
+#     python3 python3-pip git \
+#     ffmpeg libsm6 libxext6 \
+#     --no-install-recommends \
+#     && apt-get clean \
+#     && rm -rf /var/lib/apt/lists/*
+
+# # Create a working directory
+# WORKDIR /app
+
+# # Copy requirements and install dependencies
+# COPY requirements.txt .
+# RUN pip3 install --no-cache-dir -r requirements.txt
+
+# # Copy the application code
+# COPY app.py .
+
+# # Set environment variables
+# ENV PYTHONUNBUFFERED=1
+# ENV GRADIO_SERVER_NAME=0.0.0.0
+# ENV GRADIO_SERVER_PORT=7860
+
+# # Expose the port
+# EXPOSE 7860
+
+# # Command to run the application
+# CMD ["python3", "app.py"]
 FROM nvidia/cuda:12.1.1-runtime-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive
 
@@ -9,6 +40,9 @@ RUN apt-get update && apt-get install -y \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
+# Create a non-root user
+RUN useradd -m -u 1000 user
+
 # Create a working directory
 WORKDIR /app
 
@@ -17,15 +51,28 @@ COPY requirements.txt .
 RUN pip3 install --no-cache-dir -r requirements.txt
 
 # Copy the application code
-COPY app.py .
+COPY app.py download_model.py start.sh ./
+RUN chmod +x start.sh
+
+# Create cache directories with proper permissions
+RUN mkdir -p /tmp/transformers_cache /tmp/hf_home /tmp/hf_hub_cache /tmp/matplotlib_config
+RUN chown -R user:user /app /tmp/transformers_cache /tmp/hf_home /tmp/hf_hub_cache /tmp/matplotlib_config
+RUN chmod 777 /tmp/transformers_cache /tmp/hf_home /tmp/hf_hub_cache /tmp/matplotlib_config
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
 ENV GRADIO_SERVER_NAME=0.0.0.0
 ENV GRADIO_SERVER_PORT=7860
+ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+ENV HF_HOME=/tmp/hf_home
+ENV HUGGINGFACE_HUB_CACHE=/tmp/hf_hub_cache
+ENV MPLCONFIGDIR=/tmp/matplotlib_config
+
+# Switch to non-root user
+USER user
 
 # Expose the port
 EXPOSE 7860
 
 # Command to run the application
-CMD ["python3", "app.py"]
+CMD ["./start.sh"]