Update README.md
Browse files
README.md
CHANGED
@@ -50,8 +50,8 @@ model_name = "m-elio/LLaVA-NDiNO_pt_short_it"
|
|
50 |
processor = LlavaNextProcessor.from_pretrained(model_name)
|
51 |
model = LlavaNextForConditionalGeneration.from_pretrained(model_name, torch_dtype=torch.bfloat16, low_cpu_mem_usage=True, device_map="cuda")
|
52 |
|
53 |
-
|
54 |
-
image = Image.open(
|
55 |
|
56 |
conversation = [
|
57 |
{
|
@@ -68,3 +68,5 @@ print(processor.decode(output[0][inputs.input_ids.shape[1]:]))
|
|
68 |
```
|
69 |
|
70 |
## Citation
|
|
|
|
|
|
50 |
processor = LlavaNextProcessor.from_pretrained(model_name)
|
51 |
model = LlavaNextForConditionalGeneration.from_pretrained(model_name, torch_dtype=torch.bfloat16, low_cpu_mem_usage=True, device_map="cuda")
|
52 |
|
53 |
+
url = "https://www.barnorama.com/wp-content/uploads/2016/12/03-Confusing-Pictures.jpg"
|
54 |
+
image = Image.open(requests.get(url, stream=True).raw)
|
55 |
|
56 |
conversation = [
|
57 |
{
|
|
|
68 |
```
|
69 |
|
70 |
## Citation
|
71 |
+
|
72 |
+
TBD
|