tomas-gajarsky commited on
Commit
12089af
1 Parent(s): 2d095b3

Add valence arousal text box

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -49,6 +49,7 @@ def inference(path_image: str) -> Tuple:
49
 
50
  fer_dict_str = str({face.indx: face.preds["fer"].label for face in response.faces})
51
  au_dict_str = str({face.indx: face.preds["au"].other["multi"] for face in response.faces})
 
52
  deepfake_dict_str = str({face.indx: face.preds["deepfake"].label for face in response.faces})
53
  response_str = str(response)
54
 
@@ -57,7 +58,7 @@ def inference(path_image: str) -> Tuple:
57
 
58
  os.remove(path_image)
59
 
60
- out_tuple = (pil_image, fer_dict_str, au_dict_str, deepfake_dict_str, sim_dict_str_embed, sim_dict_str_verify, response_str)
61
  return out_tuple
62
 
63
 
@@ -71,6 +72,7 @@ demo=gr.Interface(
71
  [gr.Image(type="pil", label="Face Detection and 3D Landmarks"),
72
  gr.Textbox(label="Facial Expression Recognition"),
73
  gr.Textbox(label="Facial Action Unit Detection"),
 
74
  gr.Textbox(label="DeepFake Detection"),
75
  gr.Textbox(label="Cosine similarity of Face Representation Embeddings"),
76
  gr.Textbox(label="Cosine similarity of Face Verification Embeddings"),
 
49
 
50
  fer_dict_str = str({face.indx: face.preds["fer"].label for face in response.faces})
51
  au_dict_str = str({face.indx: face.preds["au"].other["multi"] for face in response.faces})
52
+ va_dict_str = str({face.indx: face.preds["va"].other for face in response.faces})
53
  deepfake_dict_str = str({face.indx: face.preds["deepfake"].label for face in response.faces})
54
  response_str = str(response)
55
 
 
58
 
59
  os.remove(path_image)
60
 
61
+ out_tuple = (pil_image, fer_dict_str, au_dict_str, va_dict_str, deepfake_dict_str, sim_dict_str_embed, sim_dict_str_verify, response_str)
62
  return out_tuple
63
 
64
 
 
72
  [gr.Image(type="pil", label="Face Detection and 3D Landmarks"),
73
  gr.Textbox(label="Facial Expression Recognition"),
74
  gr.Textbox(label="Facial Action Unit Detection"),
75
+ gr.Textbox(label="Facial Valence Arousal"),
76
  gr.Textbox(label="DeepFake Detection"),
77
  gr.Textbox(label="Cosine similarity of Face Representation Embeddings"),
78
  gr.Textbox(label="Cosine similarity of Face Verification Embeddings"),