ahmedxeno commited on
Commit
c162001
·
verified ·
1 Parent(s): 5998569

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
3
  import requests
4
  from PIL import Image
5
 
6
- processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-str")
7
- model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-str")
8
 
9
  # load image examples
10
  urls = ['https://raw.githubusercontent.com/ku21fan/STR-Fewer-Labels/main/demo_image/1.png', 'https://raw.githubusercontent.com/HCIILAB/Scene-Text-Recognition-Recommendations/main/Dataset_images/LSVT1.jpg', 'https://raw.githubusercontent.com/HCIILAB/Scene-Text-Recognition-Recommendations/main/Dataset_images/ArT2.jpg']
 
3
  import requests
4
  from PIL import Image
5
 
6
+ processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-handwritten")
7
+ model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-handwritten")
8
 
9
  # load image examples
10
  urls = ['https://raw.githubusercontent.com/ku21fan/STR-Fewer-Labels/main/demo_image/1.png', 'https://raw.githubusercontent.com/HCIILAB/Scene-Text-Recognition-Recommendations/main/Dataset_images/LSVT1.jpg', 'https://raw.githubusercontent.com/HCIILAB/Scene-Text-Recognition-Recommendations/main/Dataset_images/ArT2.jpg']