fixed something

app.py

@@ -21,9 +21,11 @@ def predict_heart_failure(input_data):
         # Predict class (0 or 1)
         prediction = "At Risk of Heart Failure" if probability >= 0.3 else "No Risk Detected"
         
+        # Return prediction, probability, and user inputs
         return {
             "Prediction": prediction,
-            "Risk Probability": round(probability, 4)
+            "Risk Probability": round(probability, 4),
+            "User Inputs": input_data
         }
     except Exception as e:
         return {"error": str(e)}
@@ -33,7 +35,7 @@ inputs = []
 for feature in features:
     inputs.append(gr.inputs.Textbox(label=feature, placeholder=f"Enter value for {feature}"))
 
-output = gr.outputs.JSON(label="Heart Failure Prediction")
+output = gr.outputs.JSON(label="Heart Failure Prediction & Inputs")
 
 interface = gr.Interface(
     fn=predict_heart_failure,