Upload 2 files

src/Dockerfile

@@ -11,10 +11,10 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt ./
 COPY src/ ./src/
 
-RUN pip3 install -r requirements.txt
+RUN pip install --upgrade pip
 
 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", "--server.enableXsrfProtection=false"]
+ENTRYPOINT ["streamlit", "run", "src/app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.enableXsrfProtection=false"]

src/requirements.txt

@@ -1,6 +1,6 @@
 torch==1.11.0
 torchvision==0.21.0
-tensorflow==2.17.1
+tensorflow==2.19.0
 streamlit==1.44.0
 pandas==2.2.3
 seaborn==0.13.2