hmarzan commited on
Commit
0653a8d
1 Parent(s): 4411475

Added resize previous to ocr processing for larger images

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -25,6 +25,8 @@ task_prompt = f"<s_{task_name}>"
25
 
26
  def predict_matricula(model, task_name, image):
27
  image = ImageOps.exif_transpose(image)
 
 
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
 
30
  model.eval()
 
25
 
26
  def predict_matricula(model, task_name, image):
27
  image = ImageOps.exif_transpose(image)
28
+ image = image.resize(size=(1280, 960),
29
+ resample=Image.Resampling.NEAREST)
30
  device = "cuda" if torch.cuda.is_available() else "cpu"
31
 
32
  model.eval()