Spaces:

Zayn
/

Image_Captioning_Using_Vision_Transformer_and_GPT-2

Runtime error

Nissan commited on Sep 17, 2022

Commit

bfd27d3

1 Parent(s): f0b9afe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ tokenizer = PreTrainedTokenizerFast.from_pretrained("distilgpt2")
 def vit2distilgpt2(img):
   pixel_values = vit_feature_extractor(images=img, return_tensors="pt").pixel_values
-  encoder_outputs = model.generate(pixel_values.to('çpu'),num_beams=5)
   generated_sentences = tokenizer.batch_decode(encoder_outputs, skip_special_tokens =True)
   return(generated_sentences[0].split('.')[0])

 def vit2distilgpt2(img):
   pixel_values = vit_feature_extractor(images=img, return_tensors="pt").pixel_values
+  encoder_outputs = model.generate(pixel_values.to('cpu'),num_beams=5)
   generated_sentences = tokenizer.batch_decode(encoder_outputs, skip_special_tokens =True)
   return(generated_sentences[0].split('.')[0])