krypton commited on
Commit
92af8f1
·
verified ·
1 Parent(s): b5351a3

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
2
+ import requests
3
+ from PIL import Image
4
+
5
+ processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
6
+ model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
7
+
8
+ # load image from the IAM dataset
9
+ url = "https://fki.tic.heia-fr.ch/static/img/a01-122-02.jpg"
10
+ image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
11
+
12
+ pixel_values = processor(image, return_tensors="pt").pixel_values
13
+ generated_ids = model.generate(pixel_values)
14
+
15
+ generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
16
+ print(generated_text)