ejschwartz commited on
Commit
7ded1cc
·
1 Parent(s): 87c3f56
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -103,9 +103,10 @@ with gr.Blocks() as demo:
103
  fun_data = get_all_dis(file.name)
104
 
105
  addrs = ["%#x" % addr for addr in fun_data.keys()]
 
106
 
107
  return {col: gr.update(visible=True),
108
- fun_dropdown: gr.Dropdown.update(choices=addrs, value=addrs[0]),
109
  all_dis_state: fun_data
110
  }
111
 
@@ -115,8 +116,8 @@ with gr.Blocks() as demo:
115
  load_results = model.fn(disassembly_str)
116
  top_k = {e['label']: e['confidence'] for e in load_results['confidences']}
117
 
118
- return {disassembly: gr.Textbox.update(value=disassembly_str),
119
- clazz: gr.Label.update(top_k),
120
  }
121
 
122
  file_widget.change(file_change_fn, file_widget, [col, fun_dropdown, all_dis_state])
 
103
  fun_data = get_all_dis(file.name)
104
 
105
  addrs = ["%#x" % addr for addr in fun_data.keys()]
106
+ default_addr = addrs[0] if addrs else None
107
 
108
  return {col: gr.update(visible=True),
109
+ fun_dropdown: gr.update(choices=addrs, value=default_addr),
110
  all_dis_state: fun_data
111
  }
112
 
 
116
  load_results = model.fn(disassembly_str)
117
  top_k = {e['label']: e['confidence'] for e in load_results['confidences']}
118
 
119
+ return {disassembly: gr.update(value=disassembly_str),
120
+ clazz: gr.update(value=top_k),
121
  }
122
 
123
  file_widget.change(file_change_fn, file_widget, [col, fun_dropdown, all_dis_state])