Update 2 files

- /yolov5/detect.py
- /yolov5/train.py

yolov5/detect.py

@@ -98,7 +98,7 @@ def run(
     screenshot = source.lower().startswith('screen')
     if is_url and is_file:
         source = check_file(source)  # download
-    if not args.name:
+    if not name:
         project_name = input('Please enter the name of your project : ')
     else:
         project_name = args.name

yolov5/train.py

@@ -512,7 +512,7 @@ def main(opt, callbacks=Callbacks()):
             opt.exist_ok, opt.resume = opt.resume, False  # pass resume to exist_ok and disable resume
         if opt.name == 'cfg':
             opt.name = Path(opt.cfg).stem  # use model.yaml as name
-        if not args.name:
+        if not opt.name:
             project_name = input('Please enter the name of your project : ')
         else:
             project_name = args.name