stepfun-ai
/

GOT-OCR2_0

Image-Text-to-Text

feature-extraction

vision-language

Model card Files Files and versions Community

ucaslcl commited on Sep 13

Commit

c5df14f

•

1 Parent(s): 41770f3

Update README.md

Files changed (1) hide show

README.md +6 -6

README.md CHANGED Viewed

@@ -34,22 +34,22 @@ model = model.eval().cuda()
 image_file = 'xxx.jpg'
 # plain texts OCR
-model.chat(tokenizer, image_file, ocr_type='ocr')
 # format texts OCR:
 # model.chat(tokenizer, image_file, ocr_type='format')
 # fine-grained OCR:
-# model.chat(tokenizer, image_file, ocr_type='ocr', ocr_box='')
-# model.chat(tokenizer, image_file, ocr_type='format', ocr_box='')
-# model.chat(tokenizer, image_file, ocr_type='ocr', ocr_color='')
-# model.chat(tokenizer, image_file, ocr_type='format', ocr_color='')
 # multi-crop OCR:
 # res = model.chat_crop(tokenizer, image_file = image_file)
 # render the formatted OCR results:
-# model.chat(tokenizer, image_file, ocr_type='format', ocr_box='', ocr_color='', render=True, save_render_file = './demo.html')
 print(res)

 image_file = 'xxx.jpg'
 # plain texts OCR
+res = model.chat(tokenizer, image_file, ocr_type='ocr')
 # format texts OCR:
 # model.chat(tokenizer, image_file, ocr_type='format')
 # fine-grained OCR:
+# res = model.chat(tokenizer, image_file, ocr_type='ocr', ocr_box='')
+# res = model.chat(tokenizer, image_file, ocr_type='format', ocr_box='')
+# res = model.chat(tokenizer, image_file, ocr_type='ocr', ocr_color='')
+# res = model.chat(tokenizer, image_file, ocr_type='format', ocr_color='')
 # multi-crop OCR:
 # res = model.chat_crop(tokenizer, image_file = image_file)
 # render the formatted OCR results:
+# res = model.chat(tokenizer, image_file, ocr_type='format', ocr_box='', ocr_color='', render=True, save_render_file = './demo.html')
 print(res)