Hugging Face's logo

Spaces:
EvanMarie
/
faces_three
Runtime error

App Files Community
EvanMarie commited on
Commit
127101e
•
1 Parent(s): 1c200b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,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_pets.predict(img)
12
  return dict(zip(categories, map(float, probability)))
13
 
14
  intf = gr.Interface(fn=classify_faces,
 
8
  categories = ['happy', 'sad', 'angry']
9
 
10
  def classify_faces(img):
11
+ prediction, index, probability = learner_faces.predict(img)
12
  return dict(zip(categories, map(float, probability)))
13
 
14
  intf = gr.Interface(fn=classify_faces,