fix: typo in interface.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -127,7 +127,7 @@ def load_model(
|
|
| 127 |
checkpoint_path = os.path.join("/tmp", model_name)
|
| 128 |
|
| 129 |
# 3. Download it to /tmp if it isn't already there
|
| 130 |
-
if not os.path.isfile(
|
| 131 |
link = pretrained_links.get(model_name)
|
| 132 |
if not link or link == "link":
|
| 133 |
raise FileNotFoundError(f"No pretrained link for {model_name}")
|
|
|
|
| 127 |
checkpoint_path = os.path.join("/tmp", model_name)
|
| 128 |
|
| 129 |
# 3. Download it to /tmp if it isn't already there
|
| 130 |
+
if not os.path.isfile(checkpoint_path):
|
| 131 |
link = pretrained_links.get(model_name)
|
| 132 |
if not link or link == "link":
|
| 133 |
raise FileNotFoundError(f"No pretrained link for {model_name}")
|