Updated image size

app.py

@@ -37,7 +37,7 @@ def inference(input_img, conf_thres, iou_thres):
     device = select_device('cpu')
     model = DetectMultiBackend(weights, device=device, dnn=False, data=data, fp16=False)
     stride, names, pt = model.stride, model.names, model.pt
-    imgsz = check_img_size(imgsz, s=stride)  # check image size
+    imgsz = check_img_size((640, 640), s=stride)  # check image size
 
     bs = 1
     # Run inference