Update app.py

app.py

@@ -98,14 +98,12 @@ def main():
 
         pred_button = st.button("Predict")
         
-        #model = load_model()
+        model = load_model()
         # label = ['Zebra', 'Horse']
         # label = np.array(label).reshape(1, -1)
         # ohe= OneHotEncoder()
         # labels = ohe.fit_transform(label).toarray()
 
-        model=tf.keras.models.load_model('model.h5', compile=False)
-
         if pred_button:
             
             outputs = model.predict(image_pred)