Spaces:

Adityaswara
/

Sign-Language-Gesture-Prediction

Sleeping

Adityaswara commited on Feb 20

Commit

8681d20

•

1 Parent(s): 4e9f39c

Update prediction.py

Files changed (1) hide show

prediction.py CHANGED Viewed

@@ -14,7 +14,7 @@ import pickle
 #load model
 def run():
     file = st.file_uploader("Upload an image", type=["jpg", "png"])
-    model = tf.keras.models.load_model('model_ann_sequential_improve.keras')
     target_size=(50, 50)
     def import_and_predict(image_data, model):

 #load model
 def run():
     file = st.file_uploader("Upload an image", type=["jpg", "png"])
+    model = tf.keras.models.load_model('model_ann_sequential_improve.h5')
     target_size=(50, 50)
     def import_and_predict(image_data, model):