Update README.md
Browse files
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(
|