xJuuzouYTx commited on
Commit
826f537
1 Parent(s): a87192b
Files changed (1) hide show
  1. 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(self._model_name, "./zips/", "./weights/")
 
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: