Spaces:
Runtime error
Runtime error
Commit
·
524171c
1
Parent(s):
f8f66e0
file
Browse files- PE_main.py +2 -2
- app.py +1 -1
PE_main.py
CHANGED
@@ -76,9 +76,9 @@ def get_version_info(pe):
|
|
76 |
return res
|
77 |
|
78 |
#extract the info for a given file using pefile
|
79 |
-
def extract_infos(
|
80 |
res = {}
|
81 |
-
pe = pefile.PE(
|
82 |
res['Machine'] = pe.FILE_HEADER.Machine
|
83 |
res['SizeOfOptionalHeader'] = pe.FILE_HEADER.SizeOfOptionalHeader
|
84 |
res['Characteristics'] = pe.FILE_HEADER.Characteristics
|
|
|
76 |
return res
|
77 |
|
78 |
#extract the info for a given file using pefile
|
79 |
+
def extract_infos(file):
|
80 |
res = {}
|
81 |
+
pe = pefile.PE(data=file)
|
82 |
res['Machine'] = pe.FILE_HEADER.Machine
|
83 |
res['SizeOfOptionalHeader'] = pe.FILE_HEADER.SizeOfOptionalHeader
|
84 |
res['Characteristics'] = pe.FILE_HEADER.Characteristics
|
app.py
CHANGED
@@ -352,7 +352,7 @@ if __name__ == "__main__":
|
|
352 |
|
353 |
# Button to extract vector
|
354 |
b_extract.click(
|
355 |
-
encode_quantize(file_input
|
356 |
inputs=[],
|
357 |
outputs=[extracted_vector],
|
358 |
)
|
|
|
352 |
|
353 |
# Button to extract vector
|
354 |
b_extract.click(
|
355 |
+
encode_quantize(file_input, eval_key, encodings),
|
356 |
inputs=[],
|
357 |
outputs=[extracted_vector],
|
358 |
)
|