Spaces:

snehalsapkale
/

demo-image-classification

Runtime error

snehalsapkale commited on Feb 29

Commit

d8dee6d

•

1 Parent(s): 53edacb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,3 +18,4 @@ if img_file_buffer is not None:
     predicted_class_idx = logits.argmax(-1).item()
     print("Predicted class:", model.config.id2label[predicted_class_idx])

     predicted_class_idx = logits.argmax(-1).item()
     print("Predicted class:", model.config.id2label[predicted_class_idx])
+    st.json(model.config.id2label[predicted_class_idx])