Update app.py

app.py

@@ -86,10 +86,22 @@ def adr_predict(x):
     # scores = list(zip(shap_values.data[0], shap_values.values[0, :, 1]))
     # # plot the first sentence's explanation
     # # plt = shap.plots.text(shap_values[0],display=False)
-    shap_scores = interpretation_function(str(x).lower())
+    
+    # shap_scores = interpretation_function(str(x).lower())
+
+    shap_values = explainer(str(x).lower(), fixed_context=1)
+    plot = shap.plots.text(shap_values[0], max_display=6, 
+                           order=shap.Explanation.abs, show_data='auto', show=False)
+
+    plt.tight_layout()
+    local_plot = plt.gcf()
+    plt.rcParams['figure.figsize'] = 6,4
+    plt.close()
+    
 
 
-    return {"Severe Reaction": float(scores.numpy()[1]), "Non-severe Reaction": float(scores.numpy()[0])}, shap_scores
+    return {"Severe Reaction": float(scores.numpy()[1]), "Non-severe Reaction": float(scores.numpy()[0])}, local_plot
+    # shap_scores
     # , word_attributions ,scores
 
 def main(prob1):
@@ -119,7 +131,8 @@ with gr.Blocks(title=title) as demo:
         #                                         "--": "darkblue",
         #                                         "-": "blue", "NA":"white"})
 
-        interpretation = gr.components.Interpretation(prob1)
+        # interpretation = gr.components.Interpretation(prob1)
+        local_plot = gr.Plot(label = 'Shap:')
 
 
     submit_btn.click(
@@ -127,12 +140,12 @@ with gr.Blocks(title=title) as demo:
         [prob1],
         [label
          # ,intp
-         ,interpretation
+         ,local_plot
         ], api_name="adr"
     )
 
     gr.Markdown("### Click on any of the examples below to see to what extent they contain resilience messaging:")
-    gr.Examples([["I have minor pain."],["I have severe pain."]], [prob1], [label,interpretation
+    gr.Examples([["I have minor pain."],["I have severe pain."]], [prob1], [label,local_plot
                                                                           ], main, cache_examples=True)
     
 demo.launch()