Update app.py

app.py

@@ -39,21 +39,8 @@ def predict(image):
 
     # Make prediction
     prediction = model.predict(img_array)
-    # Calculate prediction probability
-    prediction_probability = float(prediction[0][0])
-
-    # Check if prediction is 'Normal' or 'Pneumonia'
-    if prediction_probability <= 0.5:
-        # If 'Normal', add 70% to its probability but cap it at 1.0 (100%)
-        adjusted_probability = min(prediction_probability + 0.5, 1.0)
-        pred_class = 'Normal'
-    else:
-        # If 'Pneumonia', use the original probability
-        adjusted_probability = prediction_probability
-        pred_class = 'Pneumonia'
-
-    # Return the prediction class and the (adjusted) probability
-    return {pred_class: adjusted_probability}
+    pred_class = 'Pneumonia' if prediction[0][0] > 0.5 else 'Normal'  # Update based on your model output
+    return {pred_class: float(prediction[0][0])}
 
 iface = gr.Interface(fn=predict,
                      inputs=gr.components.Image(image_mode='RGB', label="Upload X-ray Image"),