Update app.py

app.py

@@ -17,23 +17,25 @@ def preprocess_image(image):
 
 # Prediction function
 def predict(image):
-    image_array = preprocess_image(image)
-    prediction = model.predict(image_array)[0][0]
-    label = "Pneumonia" if prediction > 0.5 else "Normal"
-    confidence = prediction if prediction > 0.5 else 1 - prediction
-    return f"{label} ({confidence * 100:.2f}% confidence)"
+    try:
+        image_array = preprocess_image(image)
+        prediction = model.predict(image_array)[0][0]
+        label = "Pneumonia" if prediction > 0.5 else "Normal"
+        confidence = prediction if prediction > 0.5 else 1 - prediction
+        return f"{label} ({confidence * 100:.2f}% confidence)"
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 # Gradio interface
 interface = gr.Interface(
     fn=predict,
-    inputs=gr.Image(type="pil"),
-    outputs="text",
-    title="Pneumonia Detection from Chest X-rays",
-    description="Upload a chest X-ray image to detect if the person has pneumonia.",
-    theme="huggingface",
+    inputs=gr.Image(type="pil", label="Upload Chest X-ray"),
+    outputs=gr.Textbox(label="Prediction"),
+    title="🫁 Pneumonia Detection from Chest X-rays",
+    description="Upload a chest X-ray image to detect if the person has pneumonia using a deep learning model.",
+    theme="default",
     allow_flagging="never"
 )
 
-# Launch the app
-if __name__ == "__main__":
-    interface.launch()
+# Important: Always call launch() outside of __main__ block on Hugging Face Spaces
+interface.launch(debug=True)