Spaces:

snehalsapkale
/

demo-image-classification

Runtime error

snehalsapkale commited on Feb 29

Commit

c88c11f

•

1 Parent(s): d8dee6d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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


18
19	predicted_class_idx = logits.argmax(-1).item()
20	print("Predicted class:", model.config.id2label[predicted_class_idx])