ejschwartz commited on
Commit
4eca714
·
1 Parent(s): d0daa31
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -119,7 +119,13 @@ with gr.Blocks() as demo:
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']}
124
 
125
  return {disassembly: gr.update(value=disassembly_str),
 
119
  print("Selected function:", selected_fun)
120
 
121
  disassembly_str = fun_data[int(selected_fun, 16)].decode("utf-8")
122
+ print("Evaluating model")
123
+ try:
124
+ load_results = model.fn(disassembly_str)
125
+ except Exception as e:
126
+ print("Error evaluating model:", e)
127
+ raise
128
+ print("Done with model")
129
  top_k = {e['label']: e['confidence'] for e in load_results['confidences']}
130
 
131
  return {disassembly: gr.update(value=disassembly_str),