Update Dockerfile

Dockerfile

@@ -14,6 +14,12 @@ RUN apt-get update && apt-get install -y \
 # Set Hugging Face cache directory to a writable location
 ENV HF_HOME /app/.cache
 
+# Set a writable config directory for Ultralytics
+ENV YOLO_CONFIG_DIR /app/.config/ultralytics
+
+# Set a writable cache directory for Matplotlib
+ENV MPLCONFIGDIR /app/.config/matplotlib
+
 # Copy all project files into the Docker image
 COPY . /app
 
@@ -24,4 +30,4 @@ RUN pip install --no-cache-dir -r requirements.txt
 EXPOSE 7860
 
 # Command to run the Streamlit app directly
-CMD ["streamlit", "run", "model.py", "--server.port", "7860", "--server.enableCORS", "false", "--server.enableXsrfProtection", "false", "--browser.gatherUsageStats", "false"]
+CMD ["streamlit", "run", "model.py", "--server.port", "7860", "--server.enableCORS", "false", "--server.enableXsrfProtection", "false", "--browser.gatherUsageStats", "false"]