Salesforce
/

instructblip-vicuna-7b

Image-Text-to-Text

image-captioning

Model card Files Files and versions

nielsr HF Staff commited on Jun 4, 2023

Commit

904cf0a

·

1 Parent(s): 60c7257

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ model.to(device)
 url = "https://raw.githubusercontent.com/salesforce/LAVIS/main/docs/_static/Confusing-Pictures.jpg"
 image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
 prompt = "What is unusual about this image?"
-inputs = processor(images=image, text=prompt, return_tensors="pt")
 outputs = model.generate(
         **inputs,

 url = "https://raw.githubusercontent.com/salesforce/LAVIS/main/docs/_static/Confusing-Pictures.jpg"
 image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
 prompt = "What is unusual about this image?"
+inputs = processor(images=image, text=prompt, return_tensors="pt").to(device)
 outputs = model.generate(
         **inputs,