fix

Dockerfile

@@ -18,4 +18,4 @@ 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.enableXsrfProtection=false"]

src/streamlit_app.py

@@ -1,5 +1,4 @@
 import io
-import os
 
 from PIL import Image
 import streamlit as st

src/utils.py

@@ -1,4 +1,3 @@
-import os
 from typing import Any, Dict, List, Optional
 
 import torch