Update Dockerfile

Dockerfile

@@ -9,6 +9,12 @@ RUN apt-get update && apt-get install -y \
 # Set working directory
 WORKDIR /app
 
+# Create a directory for Triton cache with proper permissions
+RUN mkdir -p /app/.triton && chmod 777 /app/.triton
+
+# Set environment variable to use the custom Triton cache directory
+ENV TRITON_CACHE_DIR=/app/.triton
+
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
 
@@ -22,4 +28,4 @@ COPY . .
 EXPOSE 8501
 
 # Command to run the application
-CMD ["streamlit", "run", "app.py", "--server.address", "0.0.0.0"] 
+CMD ["streamlit", "run", "app.py", "--server.address", "0.0.0.0"]