Update Dockerfile

Dockerfile

@@ -20,6 +20,15 @@ RUN mkdir -p /app/.cache/huggingface/transformers \
     && mkdir -p /app/.streamlit \
     && chmod -R 777 /app
 
+# Set Streamlit config to disable CORS and XSRF protection
+RUN echo "\
+[server]\n\
+headless = true\n\
+port = 8501\n\
+enableCORS = false\n\
+enableXsrfProtection = false\n\
+" > /app/.streamlit/config.toml
+
 # Copy files
 COPY requirements.txt ./
 COPY src/ ./src/
@@ -27,11 +36,9 @@ COPY src/ ./src/
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Expose Streamlit's default port
 EXPOSE 8501
 
-# Healthcheck for container status
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+
+
 
-# Start the Streamlit app
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]