Update README.md
Browse files
README.md
CHANGED
@@ -54,7 +54,7 @@ image = load_image("https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-L
|
|
54 |
|
55 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
56 |
|
57 |
-
model_id = "
|
58 |
processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b", do_image_splitting=False)
|
59 |
model = AutoModelForVision2Seq.from_pretrained(
|
60 |
model_id,
|
|
|
54 |
|
55 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
56 |
|
57 |
+
model_id = "cmarkea/idefics2-8b-ft-docvqa-lora"
|
58 |
processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b", do_image_splitting=False)
|
59 |
model = AutoModelForVision2Seq.from_pretrained(
|
60 |
model_id,
|