Update app.py

app.py

@@ -5,13 +5,13 @@ import cv2
 import tensorflow as tf  # Assuming you're using TensorFlow for loading your model
 
 # Load your model
-model = tf.keras.models.load_model("Brain Tumors.h5")
+model = tf.keras.models.load_model("brain_tumor.h5")
 
 def img_pred(upload):
     # Convert the Gradio input image to OpenCV format
     opencvImage = cv2.cvtColor(np.array(upload), cv2.COLOR_RGB2BGR)
-    img = cv2.resize(opencvImage, (150, 150))
-    img = img.reshape(1, 150, 150, 3)
+    img = cv2.resize(opencvImage, (128, 128))
+    img = img.reshape(1, 128, 128, 3)
     
     # Predict using the model and get confidence
     predictions = model.predict(img)[0]  # Get probabilities for each class