oddadmix commited on
Commit
c6a2a16
·
verified ·
1 Parent(s): 9e2a749

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
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