Update app.py

app.py

@@ -61,7 +61,7 @@ def draw_boxes(img, bbox, identities=None, categories=None, confidences = None,
 ##################################
 
 
-def detect(img, model):
+def detect(save_img=True):
     parser = argparse.ArgumentParser()
     parser.add_argument('--weights', nargs='+', type=str, default='yolov7.pt', help='model.pt path(s)')
     parser.add_argument('--source', type=str, default='inference/images', help='source')  # file/folder, 0 for webcam
@@ -238,11 +238,7 @@ def detect(img, model):
                     confidences = dets_to_sort[:, 4]
                 
                 im0 = draw_boxes(im0, bbox_xyxy, identities, categories, confidences, names, colors)
-
-                
-                    
-                
-                
+          
             # Print time (inference + NMS)
             print(f'{s}Done. ({(1E3 * (t2 - t1)):.1f}ms) Inference, ({(1E3 * (t3 - t2)):.1f}ms) NMS')
 
@@ -285,7 +281,7 @@ def detect(img, model):
         #print(f"Results saved to {save_dir}{s}")
 
     print(f'Done. ({time.time() - t0:.3f}s)')
-    return vid_writer
+    return img