kppkkp commited on
Commit
eb50750
1 Parent(s): 73bcc49

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -3
README.md CHANGED
@@ -1,10 +1,23 @@
1
  ---
2
  license: apache-2.0
3
  ---
4
- <h3><a href="https://github.com/LingyvKong/OneChart">OneChart: Purify the Chart Structural Extraction via One Auxiliary Token</a></h3>
5
 
6
  Jinyue Chen*, Lingyu Kong*, [Haoran Wei](https://scholar.google.com/citations?user=J4naK0MAAAAJ&hl=en), Chenglong Liu, [Zheng Ge](https://joker316701882.github.io/), Liang Zhao, [Jianjian Sun](https://scholar.google.com/citations?user=MVZrGkYAAAAJ&hl=en), Chunrui Han, [Xiangyu Zhang](https://scholar.google.com/citations?user=yuB-cfoAAAAJ&hl=en)
7
 
8
- [Github](https://github.com/LingyvKong/OneChart)
 
9
 
10
- [arxiv](https://arxiv.org/abs/2404.09987)
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+ <h1><a href="https://github.com/LingyvKong/OneChart">OneChart: Purify the Chart Structural Extraction via One Auxiliary Token</a></h1>
5
 
6
  Jinyue Chen*, Lingyu Kong*, [Haoran Wei](https://scholar.google.com/citations?user=J4naK0MAAAAJ&hl=en), Chenglong Liu, [Zheng Ge](https://joker316701882.github.io/), Liang Zhao, [Jianjian Sun](https://scholar.google.com/citations?user=MVZrGkYAAAAJ&hl=en), Chunrui Han, [Xiangyu Zhang](https://scholar.google.com/citations?user=yuB-cfoAAAAJ&hl=en)
7
 
8
+ [Github](https://github.com/LingyvKong/OneChart)
9
+ [arxiv](https://arxiv.org/abs/2404.09987)
10
 
11
+ ## Quickly try the demo using hugginface:
12
+ ```python
13
+ from transformers import AutoModel, AutoTokenizer
14
+
15
+ tokenizer = AutoTokenizer.from_pretrained('kppkkp/OneChart', trust_remote_code=True, use_fast=False, padding_side="right")
16
+ model = AutoModel.from_pretrained('kppkkp/OneChart', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda')
17
+ model = model.eval().cuda()
18
+
19
+ # input your test image
20
+ image_file = 'image.png'
21
+ res = model.chat(tokenizer, image_file, reliable_check=True)
22
+ print(res)
23
+ ```