Remove interpretation to upgrade gradio

app.py

@@ -1,17 +1,12 @@
 import gradio as gr
-import shap
-import transformers
 
 import os
 import re
 import subprocess
-import sys
 import tempfile
 
 model = gr.load("ejschwartz/oo-method-test-model-bylibrary", src="models")
 
-model_interp = transformers.pipeline("text-classification", "ejschwartz/oo-method-test-model-bylibrary")
-
 def get_all_dis(bname, addrs=None):
 
     anafile = tempfile.NamedTemporaryFile(prefix=os.path.basename(bname) + "_", suffix=".bat_ana")
@@ -89,8 +84,6 @@ with gr.Blocks() as demo:
             disassembly = gr.Textbox(label="Disassembly", lines=20)
             with gr.Column():
                 clazz = gr.Label()
-                interpret_button = gr.Button("Interpret (very slow)")
-            interpretation = gr.components.Interpretation(disassembly)
 
     example_widget = gr.Examples(
         examples=[f.path for f in os.scandir(os.path.join(os.path.dirname(__file__), "examples"))],
@@ -124,35 +117,11 @@ with gr.Blocks() as demo:
 
         return {disassembly: gr.Textbox.update(value=disassembly_str),
                 clazz: gr.Label.update(top_k),
-                # I can't figure out how to hide this
-                #interpretation: {}
         }
-    
-    # XXX: Ideally we'd use the gr.load model, which uses the huggingface
-    # inference API.  But shap library appears to use information in the
-    # transformers pipeline, and I don't feel like figuring out how to
-    # reimplement that, so we'll just use a regular transformers pipeline here
-    # for interpretation.
-    def interpretation_function(text, progress=gr.Progress(track_tqdm=True)):
-
-        progress(0, desc="Interpreting function")
-        explainer = shap.Explainer(model_interp)
-        shap_values = explainer([text])
-
-        # Dimensions are (batch size, text size, number of classes)
-        # Since we care about positive sentiment, use index 1
-        scores = list(zip(shap_values.data[0], shap_values.values[0, :, 1]))
-        # Scores contains (word, score) pairs
-
-
-        # Format expected by gr.components.Interpretation
-        return {"original": text, "interpretation": scores}
 
     file_widget.change(file_change_fn, file_widget, [col, fun_dropdown, all_dis_state])
 
-    fun_dropdown.change(function_change_fn, [fun_dropdown, all_dis_state], [disassembly, clazz, interpretation])
-
-    interpret_button.click(interpretation_function, disassembly, interpretation)
+    fun_dropdown.change(function_change_fn, [fun_dropdown, all_dis_state], [disassembly, clazz])
 
 demo.queue()
 demo.launch(server_name="0.0.0.0", server_port=7860, share=True)

requirements.txt

@@ -1,5 +1,4 @@
-gradio ~= 3.5 # gradio 4 removes interpretation :-(
+gradio
 pandas
 numpy >= 1.22.4
-shap
-transformers[torch]
+transformers