Update app.py

app.py

@@ -5,7 +5,7 @@ import tensorflow as tf
 import matplotlib
 matplotlib.use('Agg')
 
-# Carregar o modelo
+# Load the model
 model = tf.keras.models.load_model('model.h5')
 
 def preprocess_image(image):
@@ -17,7 +17,10 @@ def predict_pneumonia(image):
     img_array = preprocess_image(image)
     prediction = model.predict(img_array)[0][0]
 
-    if prediction >= 0.5:
+    # Use a more robust threshold for determining whether an image has pneumonia
+    threshold = 0.6
+
+    if prediction >= threshold:
         pneumonia_prediction = 1
     else:
         pneumonia_prediction = 0
@@ -40,7 +43,8 @@ gradio_interface = gr.Interface(
         ["NORMAL2-IM-1427-0001.jpeg"],
         ["NORMAL2-IM-1431-0001.jpeg"]
     ],
-    theme="default"
+    theme="default",
+    feedback=True
 )
 
 gradio_interface.launch()