Update README.md
Browse files
README.md
CHANGED
|
@@ -69,6 +69,16 @@ import os
|
|
| 69 |
from qwen_vl_utils import process_vision_info
|
| 70 |
|
| 71 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 72 |
prompt = "Below is the image of one page of a document, as well as some raw textual content that was previously extracted for it. Just return the plain text representation of this document as if you were reading it naturally. Do not hallucinate."
|
| 73 |
image.save("image.png")
|
| 74 |
|
|
|
|
| 69 |
from qwen_vl_utils import process_vision_info
|
| 70 |
|
| 71 |
|
| 72 |
+
|
| 73 |
+
model_name = "oddadmix/Qari-0.1-VL-2B-Instruct"
|
| 74 |
+
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
| 75 |
+
model_name,
|
| 76 |
+
torch_dtype="auto",
|
| 77 |
+
device_map="auto"
|
| 78 |
+
)
|
| 79 |
+
processor = AutoProcessor.from_pretrained(model_name)
|
| 80 |
+
max_tokens = 2000
|
| 81 |
+
|
| 82 |
prompt = "Below is the image of one page of a document, as well as some raw textual content that was previously extracted for it. Just return the plain text representation of this document as if you were reading it naturally. Do not hallucinate."
|
| 83 |
image.save("image.png")
|
| 84 |
|