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 |
)
|