Update Dockerfile

Dockerfile

@@ -11,6 +11,13 @@ RUN mkdir -p /workspace/huggingface_cache && \
 # Set environment variable for Hugging Face cache
 ENV HF_HOME=/workspace/huggingface_cache
 
+# Create and set permissions for the /.triton directory to avoid permission issues
+RUN mkdir -p /root/.triton && chmod -R 777 /root/.triton
+ENV TRITON_CACHE_DIR=/root/.triton
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y git
+
 # Copy the requirements file into the image
 COPY requirements.txt requirements.txt
 
@@ -25,3 +32,4 @@ EXPOSE 7860
 
 # Command to run your application
 CMD ["python", "app.py"]
+