Update app.py

app.py

@@ -104,7 +104,13 @@ def predict(fixed_acidity, volatile_acidity, citric_acid, residual_sugar,
     
     # Make predictions
     prediction = pipeline.predict(df)
-    return {"Quality Prediction": prediction[0]}
+    probabilities = pipeline.predict_proba(df)
+    # Prepare the result
+    result = {
+        "Predicted Quality": int(prediction[0]),  # Convert NumPy value to a Python int
+        "Class Probabilities": {str(i): prob for i, prob in enumerate(probabilities[0])}
+    }
+    return result
 
 # Define input components for Gradio
 inputs = [