Spaces:

cohoydwd
/

troc

Runtime error

cohoydwd commited on Oct 3, 2023

Commit

661a380

•

1 Parent(s): 93709dc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,8 +5,8 @@ import requests
 #import torch
 # Load the processor and model
-processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-printed')
-model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-printed')
 def generate_text(input_image):
     # Convert Gradio input image to PIL Image

 #import torch
 # Load the processor and model
+processor = TrOCRProcessor.from_pretrained('microsoft/trocr-large-stage1')
+model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-large-stage1')
 def generate_text(input_image):
     # Convert Gradio input image to PIL Image