atasoglu
/

vit-base-patch16-224-turkish-gpt2

vision-encoder-decoder

image-captioning

Model card Files Files and versions

atasoglu commited on Apr 6, 2024

Commit

d722476

·

verified ·

1 Parent(s): 24a36dc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ features = feature_extractor(images=[img], return_tensors="pt")
 pixel_values = features.pixel_values.to(device)
 generated_captions = tokenizer.batch_decode(
-    model.generate(pixel_values, max_new_tokens=128),
     skip_special_tokens=True,
 )

 pixel_values = features.pixel_values.to(device)
 generated_captions = tokenizer.batch_decode(
+    model.generate(pixel_values, max_new_tokens=20),
     skip_special_tokens=True,
 )