JerryWu commited on
Commit
1db8db2
·
1 Parent(s): b69d9fc

Create colab example

Browse files
Files changed (1) hide show
  1. colab_example.py +14 -0
colab_example.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !pip install transformers
2
+ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
3
+ from PIL import Image
4
+ import requests
5
+ url = 'https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg'
6
+
7
+ processor = TrOCRProcessor.from_pretrained('JerryWu/trocr-printed-model')
8
+ model = VisionEncoderDecoderModel.from_pretrained('JerryWu/trocr-printed-model')
9
+ pixel_values = processor(images=image, return_tensors="pt").pixel_values
10
+
11
+ image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
12
+ generated_ids = model.generate(pixel_values)
13
+ generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
14
+ print(generated_text)