Update README.md

README.md

@@ -26,7 +26,7 @@ license: apache-2.0
 
 
 ## Usage
-Inference using Huggingface transformers on NVIDIA GPUs. Requirements tested on python 3.10：
+Inference using Huggingface transformers on CPU. Requirements tested on python 3.10：
 ```
 torch==2.0.1
 torchvision==0.15.2
@@ -40,9 +40,9 @@ accelerate==0.28.0
 ```python
 from transformers import AutoModel, AutoTokenizer
 
-tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
-model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True, pad_token_id=tokenizer.eos_token_id)
-model = model.eval().cuda()
+tokenizer = AutoTokenizer.from_pretrained('srimanth-d/GOT_CPU', trust_remote_code=True)
+model = AutoModel.from_pretrained('srimanth-d/GOT_CPU', trust_remote_code=True, low_cpu_mem_usage=True, use_safetensors=True, pad_token_id=tokenizer.eos_token_id)
+model = model.eval()
 
 
 # input your test image