benja20029 commited on
Commit
62748ed
1 Parent(s): c9bd713

Change of model

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. emotion_classifier_model.pkl +2 -2
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 = ('angry', 'disgusted', 'fearfull', 'happy', 'neutral', 'sad', 'surprised')
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 = ('angry', 'anxious', 'calm', 'disgusted','happy', 'sad', 'scared')
9
 
10
  def classify_images(img):
11
  pred,idx,probs = learn.predict(img)
emotion_classifier_model.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c050941e5cf8fa3907715ee4ddfa05a551418ef13cf34b16f03e4fdd9b418920
3
- size 48099371
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4415fcd88762200ab1e3ed624aafa3e87aea0819769ac22afe6b7b8c6952954f
3
+ size 46989675