massage

app.py

@@ -93,9 +93,10 @@ with gr.Blocks() as demo:
 
         disassembly_str = fun_data[int(selected_fun, 16)].decode("utf-8")
         load_results = model.fn(disassembly_str)
+        top_k = {e['label']: e['confidence'] for e in d['confidences']}
 
         return {disassembly: gr.Textbox.update(value=disassembly_str),
-                clazz: gr.Label.update(load_results)
+                clazz: gr.Label.update(top_k)
         }
 
     file_widget.change(file_change_fn, file_widget, [col, fun_dropdown, all_dis_state])