Update app.py

app.py

@@ -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])