xJuuzouYTx
commited on
Commit
•
826f537
1
Parent(s):
a87192b
[REBUILD]
Browse files- inference.py +2 -1
inference.py
CHANGED
@@ -195,7 +195,8 @@ class Inference:
|
|
195 |
|
196 |
def run(self):
|
197 |
current_dir = os.getcwd()
|
198 |
-
modelname = model.model_downloader(
|
|
|
199 |
|
200 |
model_info = model.get_model(os.path.join(current_dir, 'weights') , modelname)
|
201 |
if not model_info:
|
|
|
195 |
|
196 |
def run(self):
|
197 |
current_dir = os.getcwd()
|
198 |
+
modelname = model.model_downloader(
|
199 |
+
self._model_name, "./zips/", "./weights/")
|
200 |
|
201 |
model_info = model.get_model(os.path.join(current_dir, 'weights') , modelname)
|
202 |
if not model_info:
|