mscsasem3 commited on
Commit
70ce2a2
1 Parent(s): 9db7fed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -310,9 +310,9 @@ def extract(image):
310
 
311
  #print("Started Processing")
312
  #image = line_image
313
- pixel_values = processor(images=image, return_tensors="pt").pixel_values
314
  processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
315
  model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
 
316
  generated_ids = model.generate(pixel_values)
317
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
318
  print(generated_text)
 
310
 
311
  #print("Started Processing")
312
  #image = line_image
 
313
  processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
314
  model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
315
+ pixel_values = processor(images=image, return_tensors="pt").pixel_values
316
  generated_ids = model.generate(pixel_values)
317
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
318
  print(generated_text)