jvcanavarro commited on
Commit
d6ea22f
1 Parent(s): 434d435

Add traits description

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -12,6 +12,8 @@ TRAIT_NAMES = [
12
  "Openness",
13
  ]
14
 
 
 
15
 
16
  def get_traits(video):
17
  model = load_model()
@@ -27,7 +29,7 @@ demo = gr.Interface(
27
  inputs=gr.Video(label="Video", include_audio=True),
28
  outputs=gr.Label(num_top_classes=5, label="Results"),
29
  title="Personality Traits Prediction [Prototype]",
30
- description="Predicts the 5 psychological traits of a person using an short introduction video",
31
  thumbnail="https://cdn-icons-png.flaticon.com/512/3392/3392044.png",
32
  examples="egs",
33
  cache_examples=True,
 
12
  "Openness",
13
  ]
14
 
15
+ DESCRIPTION = "\n'Extraversion': outgoing, energetic, talkative, active, assertive, etc.\n'Neuroticism': worrying, self-pitying, unstable, tense, anxious, etc.\n'Agreeableness': sympathetic, forgiving, generous, kind, appreciative, etc.\n'Conscientiousness': responsible, organized, reliable, efficient, planful, etc.\n'Openness': artistic, curious, imaginative, insightful, original, wide interests, etc."
16
+
17
 
18
  def get_traits(video):
19
  model = load_model()
 
29
  inputs=gr.Video(label="Video", include_audio=True),
30
  outputs=gr.Label(num_top_classes=5, label="Results"),
31
  title="Personality Traits Prediction [Prototype]",
32
+ description="Predicts the 5 psychological traits of a person using an short introduction video" + DESCRIPTION,
33
  thumbnail="https://cdn-icons-png.flaticon.com/512/3392/3392044.png",
34
  examples="egs",
35
  cache_examples=True,