Update Dockerfile

Dockerfile

@@ -2,11 +2,12 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Set HOME to a writable location
+# Set up environment variables for writable directories
 ENV HOME=/app
-ENV MPLCONFIGDIR=/app/.config/matplotlib
-ENV TORCH_HOME=/app/.torch
-ENV STREAMLIT_HOME=/app/.streamlit
+ENV MPLCONFIGDIR=/tmp/.matplotlib
+ENV TORCH_HOME=/tmp/.torch
+ENV STREAMLIT_HOME=/tmp/.streamlit
+ENV PYTHONPATH=$PYTHONPATH:/app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \