Update app.py

app.py

@@ -21,7 +21,7 @@ from utils.torch_utils import select_device, load_classifier, time_synchronized,
 
 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
+parser.add_argument('--source', type=str, default='inference/', help='source')  # file/folder, 0 for webcam
 parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
 parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
 parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
@@ -45,7 +45,7 @@ opt = parser.parse_args()
 
 
 def detect(img):
-    img.save("./inference/images/test.jpg")
+    img.save("inference/test.jpg")
     source, weights, view_img, save_txt, imgsz, trace = opt.source, opt.weights, opt.view_img, opt.save_txt, opt.img_size, opt.trace
     save_img = True  # save inference images
     webcam = source.isnumeric() or source.endswith('.txt') or source.lower().startswith(