paran3xus commited on
Commit
84337b0
·
verified ·
1 Parent(s): 173edeb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -3
README.md CHANGED
@@ -1,3 +1,23 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+ # Typst Euqation OCR Model
5
+ A pretrained TrOCR model for Typst equation OCR tasks.
6
+
7
+ ## Usage
8
+ ```python
9
+ from PIL import Image
10
+ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
11
+
12
+ processor = TrOCRProcessor.from_pretrained("paran3xus/typst_eq_ocr")
13
+ model = VisionEncoderDecoderModel.from_pretrained('paran3xus/typst_eq_ocr')
14
+
15
+ image_fps = [
16
+ 'testimg/1.png',
17
+ ]
18
+ images = [Image.open(fp).convert('RGB') for fp in image_fps]
19
+ pixel_values = processor(images=images, return_tensors="pt").pixel_values
20
+ generated_ids = model.generate(pixel_values)
21
+ generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)
22
+ [print(i) for i in generated_text]
23
+ ```