xtlyxt commited on
Commit
da3ed73
·
verified ·
1 Parent(s): 3dbcc1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from PIL import Image
3
  from transformers import pipeline
4
 
5
  # Create an image classification pipeline with scores
6
- pipe = pipeline("image-classification", model="trpakov/vit-face-expression", return_all_scores=True)
7
 
8
  # Define emotion labels
9
  emotion_labels = ["Happy", "Sad", "Angry", "Surprised", "Neutral"]
 
3
  from transformers import pipeline
4
 
5
  # Create an image classification pipeline with scores
6
+ pipe = pipeline("image-classification", model="trpakov/vit-face-expression", top_k=None)
7
 
8
  # Define emotion labels
9
  emotion_labels = ["Happy", "Sad", "Angry", "Surprised", "Neutral"]