Zled commited on
Commit
8add0d0
1 Parent(s): 2b8fefd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def get_emotion(microphone, file_upload, task):
24
 
25
  logits = model(**test).logits
26
  predicted_ids = torch.argmax(logits, dim=-1)
27
- labels = [model.config.id2label[_id] for _id in predicated_ids.tolist()]
28
  return labels
29
 
30
  demo = gr.Blocks()
 
24
 
25
  logits = model(**test).logits
26
  predicted_ids = torch.argmax(logits, dim=-1)
27
+ labels = [model.config.id2label[_id] for _id in predicted_ids.tolist()]
28
  return labels
29
 
30
  demo = gr.Blocks()