Spaces:

KaburaJ
/

binary_image_classification_app

Runtime error

KaburaJ commited on Jan 5, 2023

Commit

2f7cbf4

1 Parent(s): 3dffb6e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -54,8 +54,8 @@ def load_model():
     if not os.path.isfile('model.h5'):
         subprocess.run(['curl --output model.h5 "https://github.com/KaburaJ/Binary-Image-classification/blob/main/ZebraHorse/CNN%20Application/model.h5"'], shell=True)
-    model = tf.keras.models.load_model('model.h5', compile=False)
-    return model
 # def load_model():
 #     # Load the model architecture
 #     with open('model.json', 'r') as f:

     if not os.path.isfile('model.h5'):
         subprocess.run(['curl --output model.h5 "https://github.com/KaburaJ/Binary-Image-classification/blob/main/ZebraHorse/CNN%20Application/model.h5"'], shell=True)
+        model = tf.keras.models.load_model('model.h5', compile=False)
+        return model
 # def load_model():
 #     # Load the model architecture
 #     with open('model.json', 'r') as f: