-show link image

image.py

@@ -6,7 +6,7 @@ from PIL import Image
 # prepare image + question
 url = "https://i.imgur.com/qs0CxjE_d.webp?maxwidth=760&fidelity=grand"
 image = Image.open(requests.get(url, stream=True).raw)
-text = "How many cars are there?"
+text = "What is the two color's car ?"
 
 processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
@@ -20,4 +20,5 @@ logits = outputs.logits
 idx = logits.argmax(-1).item()
 st.write("Predicted answer:", model.config.id2label[idx])
 print("question asked:", text)
+print("image link:", url)
 print("Predicted answer:", model.config.id2label[idx])