VaultChem commited on
Commit
a651d7e
β€’
1 Parent(s): d2c0e70
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -321,6 +321,7 @@ if __name__ == "__main__":
321
 
322
  #load
323
  malware_data = load_data_from_file('malware_features.txt')
 
324
 
325
  gr.Markdown("# πŸ“€ Step 2: Upload a file for scanning")
326
  gr.Markdown("## Client side")
@@ -389,7 +390,7 @@ if __name__ == "__main__":
389
 
390
  c_extract.click(
391
  fn=encrypt_encoded_quantize,
392
- inputs =[malware_data, user_id_input, eval_key_input],
393
  outputs=[encrypted_file],
394
  )
395
 
 
321
 
322
  #load
323
  malware_data = load_data_from_file('malware_features.txt')
324
+ malware_data_state = gr.State(value=malware_data)
325
 
326
  gr.Markdown("# πŸ“€ Step 2: Upload a file for scanning")
327
  gr.Markdown("## Client side")
 
390
 
391
  c_extract.click(
392
  fn=encrypt_encoded_quantize,
393
+ inputs =[malware_data_state, user_id_input, eval_key_input],
394
  outputs=[encrypted_file],
395
  )
396