Update app.py

app.py

@@ -93,12 +93,12 @@ def main():
        
         left_column, right_column = st.columns(2)
         left_column.image(image_file, caption="Uploaded image", use_column_width=True)
-        image = image_transformation(image=Image.open(image_file))
+        image_pred = image_transformation(image=Image.open(image_file))
         
 
         pred_button = st.button("Predict")
         
-        model = load_model()
+        #model = load_model()
         # label = ['Zebra', 'Horse']
         # label = np.array(label).reshape(1, -1)
         # ohe= OneHotEncoder()
@@ -106,7 +106,7 @@ def main():
 
         if pred_button:
             
-            outputs = model.predict(image)
+            outputs = model.predict(image_pred)
             #_, y_hat = outputs.max(1)
             # predicted_idx = str(y_hat.item())
             # right_column.title("Prediction")