Update app.py

app.py

@@ -71,12 +71,9 @@ def predict_braintumor(img_input):
     pred = braintumor_model.predict(img_processed)
 
     # Handle binary decision
-    confidence = pred[0][0]
+    pred = 1 if pred[0] >= 0.5 else 0
 
-    if confidence >= 0.5:
-        return "Brain Tumor Not Found!"
-    else:
-        return "Brain Tumor Found!"
+    return pred
 
 def main():
     st.title("Brain Tumor Prediction App")
@@ -107,9 +104,12 @@ def main():
             # Make prediction for Gradio (direct input of image)
             result = predict_braintumor(uploaded_img)
 
+        # Handle binary decision
+        result_text = "Brain Tumor Found!" if result == 1 else "Brain Tumor Not Found!"
+        
         # Display prediction
         st.subheader("Prediction:")
-        st.success(result)
+        st.success(result_text)
 
 if __name__ == "__main__":
     # Streamlit part
@@ -117,5 +117,11 @@ if __name__ == "__main__":
         main()
 
     # Gradio part
-    iface = gr.Interface(fn=predict_braintumor, inputs="image", outputs="text")
+    iface = gr.Interface(
+        fn=predict_braintumor,
+        inputs="image",
+        outputs="text",
+        examples=[["examples/1 no.jpg"]],
+        enable_queue=True
+    )
     iface.launch()