nutellalpb commited on
Commit
1296e0c
·
verified ·
1 Parent(s): ccffcc4

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -30
app.py DELETED
@@ -1,30 +0,0 @@
1
- from transformers import AutoModel, AutoTokenizer
2
-
3
- tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
4
- 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)
5
- model = model.eval().cuda()
6
-
7
-
8
- # input your test image
9
- image_file = 'car.jpg'
10
-
11
- # plain texts OCR
12
- res = model.chat(tokenizer, image_file, ocr_type='ocr')
13
-
14
- # format texts OCR:
15
- # res = model.chat(tokenizer, image_file, ocr_type='format')
16
-
17
- # fine-grained OCR:
18
- # res = model.chat(tokenizer, image_file, ocr_type='ocr', ocr_box='')
19
- # res = model.chat(tokenizer, image_file, ocr_type='format', ocr_box='')
20
- # res = model.chat(tokenizer, image_file, ocr_type='ocr', ocr_color='')
21
- # res = model.chat(tokenizer, image_file, ocr_type='format', ocr_color='')
22
-
23
- # multi-crop OCR:
24
- # res = model.chat_crop(tokenizer, image_file, ocr_type='ocr')
25
- # res = model.chat_crop(tokenizer, image_file, ocr_type='format')
26
-
27
- # render the formatted OCR results:
28
- # res = model.chat(tokenizer, image_file, ocr_type='format', render=True, save_render_file = './demo.html')
29
-
30
- print(res)