Update app.py

app.py

@@ -92,11 +92,17 @@ def adr_predict(x):
 
     shap_values = explainer([str(x).lower()])
     plot = shap.plots.text(shap_values[0], display=False)
+    
+    local_plot = (
+        ""
+        + plot
+        + ""
+    )
 
-    plt.tight_layout()
-    local_plot = plt.gcf()
-    plt.rcParams['figure.figsize'] = 6,4
-    plt.close()
+    # plt.tight_layout()
+    # local_plot = plt.gcf()
+    # plt.rcParams['figure.figsize'] = 6,4
+    # plt.close()
     
 
 
@@ -132,7 +138,8 @@ with gr.Blocks(title=title) as demo:
         #                                         "-": "blue", "NA":"white"})
 
         # interpretation = gr.components.Interpretation(prob1)
-        local_plot = gr.Plot(label = 'Shap:')
+        local_plot = gr.HTML(label = 'Shap:')
+        # local_plot = gr.Plot(label = 'Shap:')
 
 
     submit_btn.click(