Update app.py

app.py

@@ -1,3 +1,4 @@
+import streamlit as st
 from datasets import load_dataset
 
 dataset = load_dataset('rwcuffney/pick_a_card_test', batch_size=32, shuffle=True)
@@ -31,7 +32,7 @@ for batch in dataset:
 
     # Print the predicted class labels
     predicted_labels = [dataset.features['label'].names[i] for i in predicted_classes]
-    print(predicted_labels)
+    st.write(predicted_labels)