Update code example
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ import requests
|
|
30 |
|
31 |
# load image from the IAM database
|
32 |
url = 'https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg'
|
33 |
-
image = Image.open(requests.get(url, stream=True).raw)
|
34 |
|
35 |
processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-stage1')
|
36 |
model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-stage1')
|
|
|
30 |
|
31 |
# load image from the IAM database
|
32 |
url = 'https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg'
|
33 |
+
image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
|
34 |
|
35 |
processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-stage1')
|
36 |
model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-stage1')
|