Update README.md
Browse files
README.md
CHANGED
@@ -27,7 +27,7 @@ image_file = "https://images.pexels.com/photos/757889/pexels-photo-757889.jpeg?a
|
|
27 |
raw_image = Image.open(requests.get(image_file, stream=True).raw)
|
28 |
inputs = processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
|
29 |
|
30 |
-
output = model.generate(**inputs, max_length=256)
|
31 |
print(processor.decode(output[0], skip_special_tokens=True))
|
32 |
```
|
33 |
## result
|
|
|
27 |
raw_image = Image.open(requests.get(image_file, stream=True).raw)
|
28 |
inputs = processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
|
29 |
|
30 |
+
output = model.generate(**inputs, max_length=256, temperature=0.4, do_sample=True)
|
31 |
print(processor.decode(output[0], skip_special_tokens=True))
|
32 |
```
|
33 |
## result
|