Hugging Face's logo

Spaces:
danupurnomo
/
sample_computer_vision
Sleeping

App Files Community
rachman commited on
Commit
b83d93f
·
verified ·
1 Parent(s): a49a78c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -15,7 +15,7 @@ import pickle
15
  def run():
16
  file = st.file_uploader("Upload an image", type=["jpg", "png"])
17
 
18
- model = load_model('my_model.keras', custom_objects={'KerasLayer': KerasLayer})
19
  target_size=(224, 224)
20
 
21
  def import_and_predict(image_data, model):
 
15
  def run():
16
  file = st.file_uploader("Upload an image", type=["jpg", "png"])
17
 
18
+ model = load_model('./my_model.keras', custom_objects={'KerasLayer': KerasLayer})
19
  target_size=(224, 224)
20
 
21
  def import_and_predict(image_data, model):