modified model

app.py

@@ -12,7 +12,7 @@ if file_name is not None:
     col1, col2 = st.columns(2)
     image = Image.open(file_name)
     col1.image(image, use_column_width=True)
-    pred,pred_idx,probs = model.predict(image)
+    pred,pred_idx,probs = model.predict(PILImage.create(file_name))
     col2.header("Prediction:")
     st.write('This is a ', pred, '!')
 else: