abdulsamad commited on
Commit
bcbfd59
·
1 Parent(s): ca6ad4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ if file is not None:
16
 
17
  #resized_image = image.resize((384, 384))
18
 
19
- processor = TrOCRProcessor.from_pretrained('microsoft/trocr-small-handwritten')
20
- model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-small-handwritten')
21
  pixel_values = processor(images=image, return_tensors="pt").pixel_values
22
  generated_ids = model.generate(pixel_values)
23
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
16
 
17
  #resized_image = image.resize((384, 384))
18
 
19
+ processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
20
+ model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
21
  pixel_values = processor(images=image, return_tensors="pt").pixel_values
22
  generated_ids = model.generate(pixel_values)
23
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]