Remove neutral class and switch to resnet34

app.py

@@ -11,9 +11,9 @@ def predict(img):
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 title = "Emotion Classifier"
-description = "An emotion classifier trained on AffectNet with fastai. Identifies emotions of anger, disgust, fear, happy, neutral, and sad."
+description = "An emotion classifier trained on AffectNet with fastai. Identifies emotions of anger, disgust, fear, happy, and sad."
 
-examples = ["happy.jpg", "neutral.jpg", "sad.jpg", "angry.jpg", "fear.jpg", "disgust.jpg"]
+examples = ["happy.jpg", "sad.jpg", "angry.jpg", "fear.jpg", "disgust.jpg"]
 
 gr.Interface(
         fn=predict,

export.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2657c44c57ddd966f12825333059e05a94116ff9edbb7c5b6e95cba8f5f3074c
-size 242595533
+oid sha256:ffcda749efec518645832aa25e02f2d7fbd80d541d118faf56b46dc41279cf05
+size 87893409