Update app.py

app.py

@@ -3,7 +3,7 @@ import tensorflow as tf
 import gradio as gr
 from keras.models import load_model
 # Load the model
-model = load_model('/content/drive/MyDrive/brain-tumor-mri/model/brain_tumor.keras')
+model = load_model('brain_tumor.keras')
 labels = ['glioma', 'meningioma', 'notumor', 'pituitary']
 def load_and_prepare_image(image_path, target_size=(224, 224)):
     """Load and prepare the image for prediction."""
@@ -15,7 +15,7 @@ def load_and_prepare_image(image_path, target_size=(224, 224)):
 
 def predict(image_path):
     if image_path is None:
-        return "Please upload an image.", "/content/drive/MyDrive/brain-tumor-mri/images/ii.PNG"
+        return "Please upload an image.", "ii.PNG"
 
     # Prepare the image
     img = load_and_prepare_image(image_path)
@@ -28,7 +28,7 @@ def predict(image_path):
     # Creative prediction message
     prediction_message = f'This MRI scan suggests the presence of a {predicted_label}.'
 
-    return prediction_message, '/content/drive/MyDrive/brain-tumor-mri/images/ii.PNG'  # Returning the image path for display
+    return prediction_message, 'ii.PNG'  # Returning the image path for display
 
 # Gradio interface setup with enhancements
 iface = gr.Interface(
@@ -37,7 +37,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="filepath", label="Upload MRI Image"),
     outputs=[
         gr.Textbox(label="Prediction", interactive=False, lines=2),
-        gr.Image(value="/content/drive/MyDrive/brain-tumor-mri/images/ii.PNG", label="Always ready to assist!")
+        gr.Image(value="ii.PNG", label="Always ready to assist!")
     ],
     title="🧠 Brain Tumor Classification",
     description="Upload a brain MRI image for analysis.",
@@ -63,7 +63,7 @@ iface = gr.Interface(
         color: #333;
     }
     """,
-    examples=[["/content/drive/MyDrive/brain-tumor-mri/images/Te-noTr_0002.jpg"], ["/content/drive/MyDrive/brain-tumor-mri/images/Te-meTr_0009.jpg"]],
+    examples=[["Te-noTr_0002.jpg"], ["Te-meTr_0009.jpg"]],
 )
 
 # Launch the interface