Update download_models_hf.py
Browse files- 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 |
-
|
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)
|