ddobokki
/

vision-encoder-decoder-vit-gpt2-coco-ko

vision-encoder-decoder

Inference Endpoints

Model card Files Files and versions Community

ddobokki commited on Dec 8, 2021

Commit

93c57d7

•

1 Parent(s): 0d8558b

v2

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,5 +28,5 @@ with Image.open(requests.get(url, stream=True).raw) as img:
 generated_ids = model.generate(pixel_values.to(device),num_beams=5)
 generated_text = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
->> ['고양이 두마리가 소파에 누워 있다.']
 ```

 generated_ids = model.generate(pixel_values.to(device),num_beams=5)
 generated_text = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
+>> ['고양이 두마리가 담요 위에 누워 있다.']
 ```