Update Dockerfile

Dockerfile

@@ -1,21 +1,26 @@
 FROM python:3.9-slim
-
 WORKDIR /app
 
 RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
-    && rm -rf /var/lib/apt/lists/*
+      build-essential \
+      curl \
+      software-properties-common \
+      git \
+    && rm -rf /var/lib/apt/lists/* \
+    && mkdir -p /data/.cache/huggingface /data/.streamlit \
+    && chmod -R 777 /data
+
+ENV XDG_CACHE_HOME=/data/.cache
+ENV HF_HOME=/data/.cache/huggingface
+ENV STREAMLIT_HOME=/data/.streamlit
+ENV HOME=/data
+ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
 
 COPY requirements.txt ./
 COPY src/ ./src/
-
 RUN pip3 install -r requirements.txt
 
 EXPOSE 8501
-
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.enableCORS=false","--server.enableXsrfProtection=false"]