fix

app.py

@@ -7,7 +7,7 @@ model = YOLO('custom_model.pt')
 
 def infer(path):
     img = cv2.imread(path)
-    output = model(img)
+    output = model(source=img)
     res = output[0].cpu().numpy()
     
     # Extract bbox, cls id and conf