Hugging Face's logo

Spaces:
Adityaswara
/
Sign-Language-Gesture-Prediction
Running

App Files Community
Adityaswara commited on
Commit
8681d20
•
1 Parent(s): 4e9f39c

Update prediction.py

Browse files
Files changed (1) hide show
  1. prediction.py +1 -1
prediction.py CHANGED
@@ -14,7 +14,7 @@ import pickle
14
  #load model
15
  def run():
16
  file = st.file_uploader("Upload an image", type=["jpg", "png"])
17
- model = tf.keras.models.load_model('model_ann_sequential_improve.keras')
18
  target_size=(50, 50)
19
 
20
  def import_and_predict(image_data, model):
 
14
  #load model
15
  def run():
16
  file = st.file_uploader("Upload an image", type=["jpg", "png"])
17
+ model = tf.keras.models.load_model('model_ann_sequential_improve.h5')
18
  target_size=(50, 50)
19
 
20
  def import_and_predict(image_data, model):