Update app.py

app.py

@@ -107,7 +107,8 @@ def detect_and_annotate(
     model_name: str,
     conf_threshold: float,
     iou_threshold: float,
-    simplify_polygons_option: bool
+    simplify_polygons_option: bool,
+    imgsz: Tuple[int, int] = (640, 512)
 ) -> np.ndarray:
     # Get the selected model
     model = models[model_name]
@@ -115,6 +116,7 @@ def detect_and_annotate(
     # Perform inference
     results = model.predict(
         image,
+        imgsz=imgsz,
         conf=conf_threshold,
         iou=iou_threshold
     )[0]