Update app.py

app.py

@@ -45,10 +45,12 @@ def detect(img,model):
     parser.add_argument('--name', default='exp', help='save results to project/name')
     parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
     parser.add_argument('--trace', action='store_true', help='trace model')
-    opt = parser.parse_args()
 
     if model == "yolov7":
-        img.resize((640, 640))
+        parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
+    if model == "yolov7-e6e" or "yolov7-e6":
+        parser.add_argument('--img-size', type=int, default=1280, help='inference size (pixels)')
+    opt = parser.parse_args()
     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