Update app.py

app.py

@@ -27,11 +27,19 @@ def main_func(HighBP, HighChol, CholCheck, BMI, Smoker, Stroke, HeartDiseaseorAt
     plt.tight_layout()
     local_plot = plt.gcf()
     plt.close()
+
+    # Save the SHAP plot as an image
+    plot.savefig("/home/user/app/gradio_cached_examples/shap_plot.png")
+
+    # Return the predicted label along with the path to the saved SHAP plot
+    return {"Predicted Label": "Low Chance" if prob[0][0] < 0.5 else "High Chance", 
+            "SHAP Plot": "/home/user/app/gradio_cached_examples/shap_plot.png"}
+
     
     # Return the predicted label along with the SHAP plots
-    return {"Predicted Label": "Low Chance" if prob[0][0] < 0.5 else "High Chance", 
-            "SHAP Summary Plot": "/home/user/app/gradio_cached_examples/shap_summary_plot.png", 
-            "SHAP Summary Plot (without bar)": "/home/user/app/gradio_cached_examples/shap_summary_plot_no_bar.png"}
+    #return {"Predicted Label": "Low Chance" if prob[0][0] < 0.5 else "High Chance", 
+         #   "SHAP Summary Plot": "/home/user/app/gradio_cached_examples/shap_summary_plot.png", 
+        #  "SHAP Summary Plot (without bar)": "/home/user/app/gradio_cached_examples/shap_summary_plot_no_bar.png"}
 
     
     #return {"Low Chance": float(prob[0][0]), "High Chance": 1-float(prob[0][0])}