Spaces:

parasmech
/

Image_captioning_nlpconnect

Runtime error

parasmech commited on Jun 15, 2023

Commit

6e65235

1 Parent(s): 69a383e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning"
 #url = 'https://d2gp644kobdlm6.cloudfront.net/wp-content/uploads/2016/06/bigstock-Shocked-and-surprised-boy-on-t-113798588-300x212.jpg'
 #with Image.open(requests.get(url, stream=True).raw) as img:
-     pixel_values = vit_feature_extractor(images=img, return_tensors="pt").pixel_values
 #encoder_outputs = model.generate(pixel_values.to('cpu'),num_beams=5)

 #url = 'https://d2gp644kobdlm6.cloudfront.net/wp-content/uploads/2016/06/bigstock-Shocked-and-surprised-boy-on-t-113798588-300x212.jpg'
 #with Image.open(requests.get(url, stream=True).raw) as img:
+ #    pixel_values = vit_feature_extractor(images=img, return_tensors="pt").pixel_values
 #encoder_outputs = model.generate(pixel_values.to('cpu'),num_beams=5)