Update app.py

app.py

@@ -101,10 +101,14 @@ def main():
         pred_button = st.button("Predict")
         
         model=load_model()
+        if model is None:
+            st.error("Error: Model could not be loaded")
+            return
         # label = ['Zebra', 'Horse']
         # label = np.array(label).reshape(1, -1)
         # ohe= OneHotEncoder()
         # labels = ohe.fit_transform(label).toarray()
+        
 
         if pred_button:
             outputs = model.predict(int(image_pred))