Update app.py

app.py

@@ -34,7 +34,7 @@ def main_func(CLIMATE_SCENARIO, EAL_SCORE,SOVI_SCORE,SOCIAL,ECONOMY,HOUSING_INFR
     local_plot = plt.gcf()
     plt.close()
     
-    return {"Very Low Risk": float(prob[0][0]), "Moderately Low Risk": float(prob[0][1]), "Moderate Risk": float(prob[0][2]), "Moderately High Risk": float(prob[0][3]),  "Very High Risk": float(prob[0][4])}
+    return {"Very Low Risk": float(prob[0][0]), "Moderately Low Risk": float(prob[0][1]), "Moderate Risk": float(prob[0][2]), "Moderately High Risk": float(prob[0][3]),  "Very High Risk": float(prob[0][4])}, local_plot 
 
 # Create the UI
 title = "**Climate Risk Model** 🌍"
@@ -69,6 +69,7 @@ with gr.Blocks(title=title) as demo:
             submit_btn = gr.Button("Analyze")
         with gr.Column(visible=True) as output_col:
             label = gr.Label(label = "RISK RATING")
+            local_plot = gr.Plot(label = 'Shap:')
            
         
             submit_btn.click(