From c702916495000e93809d55683f2603d9975f847a Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 6 Dec 2019 13:47:17 -0800
Subject: [PATCH] updates

---
 models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models.py b/models.py
index 6dd1aa6f..bbc1c057 100755
--- a/models.py
+++ b/models.py
@@ -436,10 +436,9 @@ def convert(cfg='cfg/yolov3-spp.cfg', weights='weights/yolov3-spp.weights'):
 
 def attempt_download(weights):
     # Attempt to download pretrained weights if not found locally
-    file = Path(weights).name
     msg = weights + ' missing, try downloading from https://drive.google.com/open?id=1LezFG5g3BCW6iYaV89B2i64cqEUZD7e0'
-    r = 1  # error value
 
+    r = 1  # error value
     if weights and not os.path.isfile(weights):
         d = {'yolov3-spp.weights': '16lYS4bcIdM2HdmyJBVDOvt3Trx6N3W2R',
              'yolov3.weights': '1uTlyDWlnaqXcsKOktP5aH_zRDbfcDp-y',
@@ -452,6 +451,7 @@ def attempt_download(weights):
              'ultralytics49.pt': '158g62Vs14E3aj7oPVPuEnNZMKFNgGyNq',
              'ultralytics68.pt': '1Jm8kqnMdMGUUxGo8zMFZMJ0eaPwLkxSG'}
 
+        file = Path(weights).name
         if file in d:
             r = gdrive_download(id=d[file], name=weights)
         else:  # download from pjreddie.com