Spaces:

KaburaJ
/

binary_image_classification_app

Runtime error

KaburaJ commited on Jan 5, 2023

Commit

d25500f

1 Parent(s): 2f7cbf4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -54,7 +54,7 @@ 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

     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)
+        tf.keras.models.load_model('model.h5', compile=False)
         return model
 # def load_model():
 #     # Load the model architecture