Spaces:
Sleeping
Sleeping
benja20029
commited on
Commit
•
bc14618
1
Parent(s):
6d41f03
Change of emotions related to the model
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ def emotion(x): return x[0].isupper()
|
|
5 |
|
6 |
learn = load_learner('emotion_classifier_model.pkl')
|
7 |
|
8 |
-
categories = ('
|
9 |
|
10 |
def classify_images(img):
|
11 |
pred,idx,probs = learn.predict(img)
|
|
|
5 |
|
6 |
learn = load_learner('emotion_classifier_model.pkl')
|
7 |
|
8 |
+
categories = ('anger', 'disgust', 'fear', 'happiness', 'neutral', 'sadness', 'surprise')
|
9 |
|
10 |
def classify_images(img):
|
11 |
pred,idx,probs = learn.predict(img)
|