Spaces:

rbarman
/

resnet50-example

Runtime error

rbarman commited on Feb 26, 2023

Commit

235c93b

1 Parent(s): d498095

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,8 @@ def predict(image):
     # model predicts one of the 1000 ImageNet classes
     predicted_label = logits.argmax(-1).item()
-    print(model.config.id2label[predicted_label])
 # setup Gradio interface
 title = "Image classifier"

     # model predicts one of the 1000 ImageNet classes
     predicted_label = logits.argmax(-1).item()
+    prediction = model.config.id2label[predicted_label]
+    return prediction
 # setup Gradio interface
 title = "Image classifier"