test_image code calling url

app.py

@@ -43,7 +43,10 @@ model = VisionEncoderDecoderModel.from_pretrained("./vit-bert")
 '''
 
 repo_name = "ydshieh/vit-gpt2-coco-en"
-test_image = "cats.jpg"
+#test_image = "cats.jpg"
+url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
+test_image = Image.open(requests.get(url, stream=True).raw)
+test_image.save("cats.png")
 feature_extractor2 = ViTFeatureExtractor.from_pretrained(repo_name)
 tokenizer = AutoTokenizer.from_pretrained(repo_name)
 model2 = VisionEncoderDecoderModel.from_pretrained(repo_name)