abdulsamad commited on
Commit
c5f7df2
·
1 Parent(s): 492e4c6

Update app.py

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