JustKiddo commited on
Commit
dbd7548
·
1 Parent(s): c0e6571

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -5,8 +5,10 @@ from transformers import AutoProcessor, AutoModelForDocumentQuestionAnswering
5
  from PIL import Image
6
 
7
  #client = InferenceClient("models/microsoft/trocr-base-handwritten")
8
- processor = AutoProcessor.from_pretrained("microsoft/trocr-base-handwritten")
9
- model = AutoModelForDocumentQuestionAnswering.from_pretrained("microsoft/trocr-base-handwritten")
 
 
10
 
11
  def sepia(input_img):
12
  sepia_filter = np.array([
 
5
  from PIL import Image
6
 
7
  #client = InferenceClient("models/microsoft/trocr-base-handwritten")
8
+ #processor = AutoProcessor.from_pretrained("microsoft/trocr-base-handwritten")
9
+ processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
10
+ #model = AutoModelForDocumentQuestionAnswering.from_pretrained("microsoft/trocr-base-handwritten")
11
+ model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
12
 
13
  def sepia(input_img):
14
  sepia_filter = np.array([