Update app.py

app.py

@@ -50,11 +50,12 @@ set_png_as_page_bg('background.webp')
 #     #CNN_class_index = json.load(open(f"{os.getcwd()}F:\Machine Learning Resources\ZebraHorse\model.json"))
 #     return model, CNN_class_index
 
-
-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)
+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: