Spaces:

Laurits
/

celebrity-recognizer

Runtime error

Laurits commited on Sep 7, 2023

Commit

39f6d6c

1 Parent(s): f5cdcac

Change output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ learn = load_learner('model.pth')
 def classify_image(img):
     celebrity_name,_,probs = learn.predict(img)
-    return {celebrity_name: probs[0]}
 image = gr.inputs.Image(shape=(224,224))
 label = gr.outputs.Label()

 def classify_image(img):
     celebrity_name,_,probs = learn.predict(img)
+    return celebrity_name
 image = gr.inputs.Image(shape=(224,224))
 label = gr.outputs.Label()