progress bar

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.analysis.typeCheckingMode": "off"
+}

app.py

@@ -49,7 +49,8 @@ def get_all_dis(bname, addrs=None):
 
     return func_dis
 
-def get_funs(f):
+def get_funs(f, progress=gr.Progress()):
+    progress(0, "Identifying functions")
     funs = get_all_dis(f.name)
     return "\n".join(("%#x" % addr) for addr in funs.keys())
 
@@ -132,9 +133,10 @@ with gr.Blocks() as demo:
     # 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):
+    def interpretation_function(text, progress=gr.Progress()):
 
         explainer = shap.Explainer(model_interp)
+        progress(0, "Generating explanations; this takes a long time")
         shap_values = explainer([text])
 
         # Dimensions are (batch size, text size, number of classes)