Update app.py

app.py

@@ -54,9 +54,10 @@ set_png_as_page_bg('background.webp')
 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
+
+    model=tf.keras.models.load_model('model.h5', compile=False)
+    return model
+
         
 # def load_model():
 #     # Load the model architecture
@@ -99,7 +100,7 @@ def main():
 
         pred_button = st.button("Predict")
         
-        model = load_model()
+        load_model()
         # label = ['Zebra', 'Horse']
         # label = np.array(label).reshape(1, -1)
         # ohe= OneHotEncoder()