Spaces:
Sleeping
Sleeping
Commit ·
d0daa31
1
Parent(s): acbef04
debug
Browse files
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']}
|