moranyanuka
/

blip-image-captioning-base-mocha

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

moranyanuka commited on Dec 19, 2023

Commit

fc6cc98

•

1 Parent(s): 38ead43

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -100,7 +100,7 @@ inputs = processor(raw_image, text, return_tensors="pt").to("cuda", torch.float1
 out = model.generate(**inputs)
 print(processor.decode(out[0], skip_special_tokens=True))
-# >>> a photography of a woman and a dog
 # unconditional image captioning
 inputs = processor(raw_image, return_tensors="pt").to("cuda", torch.float16)

 out = model.generate(**inputs)
 print(processor.decode(out[0], skip_special_tokens=True))
+# >>> a photography of a woman and her dog on the beach
 # unconditional image captioning
 inputs = processor(raw_image, return_tensors="pt").to("cuda", torch.float16)