Update app.py

app.py

@@ -37,9 +37,14 @@ st.code(type(model))
 from transformers import AutoImageProcessor
 import torch
 
+
+image = dataset["test"][0]
+st.image(image)
+
 image_processor = AutoImageProcessor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099224"))
 inputs = image_processor(image, return_tensors="pt")
 
+
 '''
 extractor = AutoFeatureExtractor.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099225")
 model = AutoModelForImageClassification.from_pretrained("rwcuffney/autotrain-pick_a_card-3726099225")