SkyNait commited on
Commit
d92a5f5
·
verified ·
1 Parent(s): c4ad39e

Update download_models_hf.py

Browse files
Files changed (1) hide show
  1. download_models_hf.py +1 -1
download_models_hf.py CHANGED
@@ -61,7 +61,7 @@ if __name__ == '__main__':
61
  'models-dir': model_dir,
62
  'layoutreader-model-dir': layoutreader_model_dir,
63
  'device-mode': 'cuda'
64
- # 'layout_model': 'doclayout_yolo'
65
  }
66
 
67
  download_and_modify_json(json_url, config_file, json_mods)
 
61
  'models-dir': model_dir,
62
  'layoutreader-model-dir': layoutreader_model_dir,
63
  'device-mode': 'cuda'
64
+ 'layout_model': 'doclayout_yolo'
65
  }
66
 
67
  download_and_modify_json(json_url, config_file, json_mods)