Spaces:

tensorgirl
/

face_emotion

Runtime error

tensorgirl commited on Mar 4

Commit

38a3ff0

•

1 Parent(s): c4667f0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,8 +125,8 @@ def fn(image):
   img = np.expand_dims(img,axis=0)
   feats = backbone.predict(img)
   pred = model_xgb.predict(feats)
-  return emotions[pred]
 demo = gr.Interface(

   img = np.expand_dims(img,axis=0)
   feats = backbone.predict(img)
   pred = model_xgb.predict(feats)
+  print(pred)
+  return emotions[pred[0]]
 demo = gr.Interface(