Update README.md

README.md

@@ -1,3 +1,17 @@
----
-license: mit
----
+---
+license: mit
+---
+from transformers import TrOCRProcessor, VisionEncoderDecoderModel
+from PIL import Image
+import requests
+
+# load image from the IAM database
+url = 'https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg'
+image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
+
+processor = TrOCRProcessor.from_pretrained('kun826/trocr_handwrite_option4')
+model = VisionEncoderDecoderModel.from_pretrained('kun826/trocr_handwrite_option4')
+pixel_values = processor(images=image, return_tensors="pt").pixel_values
+
+generated_ids = model.generate(pixel_values)
+generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]