Spaces:

JosephH
/

Gender-Detector

Runtime error

JosephH commited on Apr 9, 2023

Commit

88466ed

1 Parent(s): 64e195c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ labels = learn.dls.vocab
 def predict(img):
     img = PILImage.create(img)
     pred,pred_idx,probs = learn.predict(img.to_thumb(32,32))
-    return {labels[i]: pred
 title = "Gender Detector"
 interpretation='default'

 def predict(img):
     img = PILImage.create(img)
     pred,pred_idx,probs = learn.predict(img.to_thumb(32,32))
+    return pred
 title = "Gender Detector"
 interpretation='default'