help

app.py

@@ -35,9 +35,7 @@ and I reused some of their own code to do this.
 hf_key = os.environ["HF_TOKEN"]
 huggingface_hub.login(token=hf_key)
 
-tokenizer = AutoTokenizer.from_pretrained(
-    "bigcode/starcoderbase-3b"
-)
+tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase-3b")
 vardecoder_model = AutoModelForCausalLM.from_pretrained(
     "ejschwartz/resym-vardecoder", torch_dtype=torch.bfloat16, device_map="auto"
 )
@@ -62,12 +60,16 @@ return sub_411142(a1, a2, &v4);
 def infer(code):
 
     splitcode = code.splitlines()
-    bodyvars = [v['name'] for v in prep_decompiled.extract_comments(splitcode) if "name" in v]
-    argvars = [v['name'] for v in prep_decompiled.parse_signature(splitcode) if "name" in v]
+    bodyvars = [
+        v["name"] for v in prep_decompiled.extract_comments(splitcode) if "name" in v
+    ]
+    argvars = [
+        v["name"] for v in prep_decompiled.parse_signature(splitcode) if "name" in v
+    ]
     vars = argvars + bodyvars
-    #comments = prep_decompiled.extract_comments(splitcode)
-    #sig = prep_decompiled.parse_signature(splitcode)
-    #print(f"vars {vars}")
+    # comments = prep_decompiled.extract_comments(splitcode)
+    # sig = prep_decompiled.parse_signature(splitcode)
+    # print(f"vars {vars}")
 
     varstring = ", ".join([f"`{v}`" for v in vars])
 
@@ -118,11 +120,13 @@ def infer(code):
 demo = gr.Interface(
     fn=infer,
     inputs=[
-        gr.Textbox(lines=10, value=example),
+        gr.Textbox(lines=10, value=example, label="Hex-Rays Decompilation"),
+    ],
+    outputs=[
+        gr.Text(label="Var Decoder Output"),
+        gr.Text(label="Field Decoder Output"),
+        gr.Text(label="Generated Variable List"),
     ],
-    outputs=[gr.Text(label="Var Decoder Output"),
-             gr.Text(label="Field Decoder Output"),
-             gr.Text(label="Generated Variable List")],
-    description=description
+    description=description,
 )
 demo.launch()