Upload 7 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+src/best_model_sore.keras filter=lfs diff=lfs merge=lfs -text

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"]

requirements.txt

@@ -1,4 +1,6 @@
-streamlit
+streamlit==1.43.0
 pandas
-tensorflow
+tensorflow<2.16
 numpy
+tf-keras<2.16
+pillow

src/best_model_sore.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff2daf06f5e4c74e1f0b904fe4af30bc18f51896d5164477e9d49767415374aa
+size 144054700

src/prediction.py

@@ -3,15 +3,13 @@ import pandas as pd
 import os
 import sys
 from io import BytesIO, StringIO
-import pickle
-import json
 import tensorflow as tf
 from PIL import Image
 import numpy as np
 
 # load files
 
-model=tf.keras.models.load_model('best_model.h5')
+model=tf.keras.models.load_model('best_model_sore.keras')
 
 klas = ['baseball_cap', 'beanie_hat', 'bucket_hat', 'fedora_hat', 'flat_cap']