Update app.py

app.py

@@ -17,8 +17,13 @@ pytesseract.pytesseract.tesseract_cmd = '/usr/bin/tesseract'
 
 # Load your YOLO model
 model = YOLO('best.pt')  # Replace with your custom model if needed
+img_dim = (640, 640)
 
 def predict(image, conf_threshold, iou_threshold):
+
+    # Resizing
+    image = image.resize(img_dim)
+    
     # Convert from PIL to OpenCV format
     image = np.array(image)
     image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)