ejschwartz commited on
Commit
d0daa31
·
1 Parent(s): acbef04
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -7,6 +7,8 @@ import tempfile
7
 
8
  model = gr.load("ejschwartz/oo-method-test-model-bylibrary", src="models")
9
 
 
 
10
  def get_all_dis(bname, addrs=None):
11
 
12
  anafile = tempfile.NamedTemporaryFile(prefix=os.path.basename(bname) + "_", suffix=".bat_ana")
@@ -93,6 +95,8 @@ with gr.Blocks() as demo:
93
 
94
  def file_change_fn(file, progress=gr.Progress()):
95
 
 
 
96
  if file is None:
97
  return {col: gr.update(visible=False),
98
  all_dis_state: None}
@@ -112,6 +116,8 @@ with gr.Blocks() as demo:
112
 
113
  def function_change_fn(selected_fun, fun_data):
114
 
 
 
115
  disassembly_str = fun_data[int(selected_fun, 16)].decode("utf-8")
116
  load_results = model.fn(disassembly_str)
117
  top_k = {e['label']: e['confidence'] for e in load_results['confidences']}
 
7
 
8
  model = gr.load("ejschwartz/oo-method-test-model-bylibrary", src="models")
9
 
10
+ print("Model loaded")
11
+
12
  def get_all_dis(bname, addrs=None):
13
 
14
  anafile = tempfile.NamedTemporaryFile(prefix=os.path.basename(bname) + "_", suffix=".bat_ana")
 
95
 
96
  def file_change_fn(file, progress=gr.Progress()):
97
 
98
+ print("File change")
99
+
100
  if file is None:
101
  return {col: gr.update(visible=False),
102
  all_dis_state: None}
 
116
 
117
  def function_change_fn(selected_fun, fun_data):
118
 
119
+ print("Selected function:", selected_fun)
120
+
121
  disassembly_str = fun_data[int(selected_fun, 16)].decode("utf-8")
122
  load_results = model.fn(disassembly_str)
123
  top_k = {e['label']: e['confidence'] for e in load_results['confidences']}