Spaces:

wijayaa65
/

gc7

Sleeping

wijayaa65 commited on May 21

Commit

593fc20

•

1 Parent(s): d302393

Upload 7 files

Files changed (1) hide show

prediction.py CHANGED Viewed

@@ -7,13 +7,13 @@ import tensorflow as tf
 from tensorflow.keras.models import load_model
 #load model
-model = load_model('ANNmodel.keras')
 def run():
     file = st.file_uploader("Upload an image", type=["jpg", "png"])
-    model = load_model('my_model.keras', custom_objects={'KerasLayer': KerasLayer})
     target_size=(360, 360)
     def import_and_predict(image_data, model):

 from tensorflow.keras.models import load_model
 #load model
 def run():
     file = st.file_uploader("Upload an image", type=["jpg", "png"])
+    model = load_model('ANNmodel.keras', custom_objects={'KerasLayer': KerasLayer})
     target_size=(360, 360)
     def import_and_predict(image_data, model):