AdithyaSK commited on
Commit
63d8ce8
·
verified ·
1 Parent(s): 19170e3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -1
README.md CHANGED
@@ -11,4 +11,23 @@ tags:
11
  license: apache-2.0
12
  ---
13
 
14
- GOT OCR v1 hi
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  license: apache-2.0
12
  ---
13
 
14
+
15
+ Nayana_base_combined_v1
16
+
17
+ ```
18
+ from transformers import AutoModel, AutoTokenizer
19
+ from peft import PeftModel, PeftConfig, AutoPeftModelForCausalLM
20
+ from transformers import AutoModelForCausalLM
21
+ import torch
22
+
23
+ tokenizer = AutoTokenizer.from_pretrained('v1v1d/Nayana_base_combined', trust_remote_code=True , torch_dtype=torch.float16)
24
+ model = AutoModel.from_pretrained('v1v1d/Nayana_base_combined', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True, pad_token_id=tokenizer.eos_token_id , torch_dtype=torch.float16)
25
+
26
+
27
+ model = model.eval().cuda()
28
+
29
+ image_file = 'hindi.png'
30
+ res = model.chat(tokenizer, image_file, ocr_type='ocr' , render=True, stream_flag = True)
31
+
32
+ print(res)
33
+ ```