Added resize previous to ocr processing for larger images
Browse files
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()
|