Spaces:

Laurits
/

celebrity-recognizer

Runtime error

Laurits commited on Sep 7, 2023

Commit

224c31d

1 Parent(s): 1ad3ceb

yes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import gradio as gr
 learn = load_learner('model.pth')
 def classify_image(img):
-    name, prob = celebrity_name,_,probs = learn.predict(img)
     return {name: prob}
 image = gr.inputs.Image(shape=(224,224))

 learn = load_learner('model.pth')
 def classify_image(img):
+    celebrity_name,_,probs = learn.predict(img)
     return {name: prob}
 image = gr.inputs.Image(shape=(224,224))