Update app.py

app.py

@@ -8,17 +8,20 @@ model = load_model('brain_tumor_model.h5')
 
 # Function to process image and make predictions
 def predict_image(image):
-    # Resize the image
+    # Resize the image to match model input
     img = image.resize((128, 128))
 
     # Convert the image to a NumPy array
     img = np.array(img)
 
-    # Check if the image has 3 color channels
+    # Ensure the image has 3 color channels (RGB)
     if img.shape == (128, 128):  # If grayscale, convert to RGB
         img = np.stack((img,) * 3, axis=-1)
 
-    # Add a batch dimension
+    # Normalize the image (if your model requires normalization)
+    img = img / 255.0
+
+    # Add a batch dimension (for model input)
     img = np.expand_dims(img, axis=0)
 
     # Make the prediction
@@ -28,22 +31,15 @@ def predict_image(image):
     predicted_class = np.argmax(prediction)
     confidence = np.max(prediction)
 
-    # Return the results
+    # Return the result with confidence level
     if predicted_class == 0:
         return f'No tumor detected. Confidence: {confidence:.2f}'
     else:
         return f'Tumor detected. Confidence: {confidence:.2f}'
 
-# Create custom CSS for background color
-#css = """
-#body {
-#    background-color: dark;
-#}
-#"""
-
 # Provide example images (these should be paths to valid images)
 examples = [
-    ["https://huggingface.co/spaces/tmafantiri/braintumourdetector/blob/main/images/no/1%20no.jpeg"],  # Replace with actual example image paths
+    ["https://huggingface.co/spaces/tmafantiri/braintumourdetector/blob/main/images/no/1%20no.jpeg"],  
     ["https://huggingface.co/spaces/tmafantiri/braintumourdetector/blob/main/images/yes/Y1.jpg"],
     ["https://huggingface.co/spaces/tmafantiri/braintumourdetector/blob/main/images/no/15%20no.jpg"],
     ["https://huggingface.co/spaces/tmafantiri/braintumourdetector/blob/main/images/yes/Y104.jpg"]
@@ -56,12 +52,10 @@ iface = gr.Interface(
     outputs=gr.Textbox(),
     title="Brain Tumor Detection AI App",
     description="Upload an image to detect brain tumors.",
-    #css=css,  # Apply the custom background color
     theme="monochrome",  # Apply a dark theme to the interface
     flagging_options=["Incorrect Diagnosis", "Image Not Clear", "Model Error"],  # Add flagging options
-    examples=examples  # Include 4 example images
+    examples=examples  # Include example images
 )
 
 # Launch the interface
-iface.launch(share=True,debug=True)
-
+iface.launch(share=True, debug=True)