mariotawfik commited on
Commit
524171c
·
1 Parent(s): f8f66e0
Files changed (2) hide show
  1. PE_main.py +2 -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(fpath):
80
  res = {}
81
- pe = pefile.PE(fpath)
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.file, eval_key, encodings),
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
  )