mariotawfik commited on
Commit
29cc13d
1 Parent(s): 25b9caf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -192,9 +192,9 @@ def decrypt_prediction(user_id):
192
  predictions = fhe_api.deserialize_decrypt_dequantize(encrypted_prediction)
193
 
194
  if(predictions[0][0] >= 0.5):
195
- return "Safe file"
196
  else:
197
- return "Malware"
198
 
199
 
200
 
 
192
  predictions = fhe_api.deserialize_decrypt_dequantize(encrypted_prediction)
193
 
194
  if(predictions[0][0] >= 0.5):
195
+ return "Safe file", predictions
196
  else:
197
+ return "Malware", predictions
198
 
199
 
200