EvanMarie commited on
Commit
20502e3
Β·
1 Parent(s): c3ecefa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
 
6
 
7
  learner_faces = load_learner('faces_resenet50_78.pk1')
8
- categories = ['happy', 'sad', 'angry']
9
 
10
  def classify_faces(img):
11
  prediction, index, probability = learner_faces.predict(img)
 
5
 
6
 
7
  learner_faces = load_learner('faces_resenet50_78.pk1')
8
+ categories = ['sad', 'happy', 'angry']
9
 
10
  def classify_faces(img):
11
  prediction, index, probability = learner_faces.predict(img)