Update app.py

app.py

@@ -37,7 +37,7 @@ def predict(img) -> Tuple[Dict,float] :
   image =  load_and_prep_imgg(img)
   image = Image.open(image)  
 
-  pred_img = model.predict(tf.expand_dims(img, axis=0))
+  pred_img = model.predict(tf.expand_dims(image, axis=0))
   pred_class = class_names[pred_img.argmax()]
   st.write(f"Predicted brain tumor is: {pred_class} with probability: {pred_img.max():.2f}")