rushai-dev commited on
Commit
37caca0
·
1 Parent(s): 26b5dfd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -0
README.md CHANGED
@@ -1,3 +1,25 @@
1
  ---
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+
5
+ ```python
6
+ from transformers import TrOCRProcessor, AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
7
+
8
+ encode = 'rushai-dev/THAI-TrOCR'
9
+ decode = "xlm-roberta-base"
10
+
11
+ tokenizer = AutoTokenizer.from_pretrained(decode)
12
+ feature_extractor = ViTFeatureExtractor.from_pretrained(encode)
13
+ processor = TrOCRProcessor(feature_extractor=feature_extractor, tokenizer=tokenizer)
14
+ model = VisionEncoderDecoderModel.from_pretrained(encode)
15
+ ```
16
+
17
+ ```python
18
+ from PIL import Image
19
+ image = Image.open("xxxxxxx.png").convert("RGB")
20
+ image
21
+ pixel_values = processor(image, return_tensors="pt").pixel_values
22
+ generated_ids = model.generate(pixel_values)
23
+ generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
24
+ generated_text
25
+ ```