Update app.py

app.py

@@ -106,8 +106,8 @@ def main():
         # labels = ohe.fit_transform(label).toarray()
 
         if pred_button:
-            image_prediction(image)
-            outputs = model.predict(image)
+            image_prediction(image=image, model=model)
+            outputs = model.predict(image_prediction)
             _, y_hat = outputs.max(1)
             predicted_idx = str(y_hat.item())
             right_column.title("Prediction")