Spaces:

JosephH
/

Gender-Detector

Runtime error

JosephH commited on Apr 9, 2023

Commit

fbbdeff

1 Parent(s): 2d9e36c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ learn = load_learner('export.pkl')
 labels = learn.dls.vocab
 def predict(img):
     img = PILImage.create(img)
-    pred,pred_idx,probs = learn.predict(img.to_thumb(64,64))
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 title = "Gender Detector"

 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]: float(probs[i]) for i in range(len(labels))}
 title = "Gender Detector"