Spaces:

rachman
/

sample_computer_vision

Sleeping

rachman commited on May 26

Commit

3b47a75

verified ·

1 Parent(s): b83d93f

Update src/streamlit_app.py

Files changed (1) hide show

src/streamlit_app.py CHANGED Viewed

@@ -15,7 +15,7 @@ import pickle
 def run():
     file = st.file_uploader("Upload an image", type=["jpg", "png"])
-    model = load_model('./my_model.keras', custom_objects={'KerasLayer': KerasLayer})
     target_size=(224, 224)
     def import_and_predict(image_data, model):

 def run():
     file = st.file_uploader("Upload an image", type=["jpg", "png"])
+    model = load_model('src/my_model.keras', custom_objects={'KerasLayer': KerasLayer})
     target_size=(224, 224)
     def import_and_predict(image_data, model):