Spaces:

KaburaJ
/

binary_image_classification_app

Runtime error

KaburaJ commited on Jan 5, 2023

Commit

d89adef

1 Parent(s): 9358ae1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,8 +55,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():

     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():