tomas-gajarsky commited on
Commit
e2c48d1
1 Parent(s): afda8c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import json
2
  import operator
3
  import gradio as gr
@@ -43,6 +44,8 @@ def inference(path_image: str) -> Tuple:
43
  sim_dict_str_embed = gen_sim_dict_str(response, pred_name="embed", index=0)
44
  sim_dict_str_verify = gen_sim_dict_str(response, pred_name="verify", index=0)
45
 
 
 
46
  out_tuple = (pil_image, fer_dict_str, deepfake_dict_str, sim_dict_str_embed, sim_dict_str_verify, response_str)
47
  return out_tuple
48
 
 
1
+ import os
2
  import json
3
  import operator
4
  import gradio as gr
 
44
  sim_dict_str_embed = gen_sim_dict_str(response, pred_name="embed", index=0)
45
  sim_dict_str_verify = gen_sim_dict_str(response, pred_name="verify", index=0)
46
 
47
+ os.remove(path_image)
48
+
49
  out_tuple = (pil_image, fer_dict_str, deepfake_dict_str, sim_dict_str_embed, sim_dict_str_verify, response_str)
50
  return out_tuple
51