Update code example

This commit is contained in:
Niels Rogge 2021-12-06 17:58:57 +00:00 committed by huggingface-web
parent 0aa347229f
commit 554a6621f6
1 changed files with 1 additions and 1 deletions

View File

@ -30,7 +30,7 @@ import requests
# load image from the IAM database # load image from the IAM database
url = 'https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg' url = 'https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg'
image = Image.open(requests.get(url, stream=True).raw) image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten') processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten') model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')