Image-Text-to-Text
English
weizhiwang commited on
Commit
c4db037
·
verified ·
1 Parent(s): 51c6cbd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -44,8 +44,7 @@ vlm.to(device, dtype=torch.bfloat16)
44
  image_url = "https://huggingface.co/adept/fuyu-8b/resolve/main/bus.png"
45
  # image = Image.open(requests.get(image_url, stream=True).raw).convert("RGB")
46
  image = [vlm.vision_backbone.image_transform(Image.open(requests.get(image_url, stream=True).raw).convert("RGB")).unsqueeze(0)]
47
- user_prompt = '<image>' + '
48
- ' + "Describe the image."
49
 
50
  # Generate!
51
  generated_text = vlm.generate_batch(
 
44
  image_url = "https://huggingface.co/adept/fuyu-8b/resolve/main/bus.png"
45
  # image = Image.open(requests.get(image_url, stream=True).raw).convert("RGB")
46
  image = [vlm.vision_backbone.image_transform(Image.open(requests.get(image_url, stream=True).raw).convert("RGB")).unsqueeze(0)]
47
+ user_prompt = '<image>\nDescribe the image."
 
48
 
49
  # Generate!
50
  generated_text = vlm.generate_batch(