2015-11-15 14:58:21 +00:00
|
|
|
import plac
|
2016-10-23 17:46:28 +00:00
|
|
|
import sputnik
|
|
|
|
|
|
2016-03-24 10:19:43 +00:00
|
|
|
from ..download import download
|
2016-10-23 17:46:28 +00:00
|
|
|
from .. import about
|
2015-01-30 07:04:01 +00:00
|
|
|
|
2015-01-17 05:21:17 +00:00
|
|
|
|
2015-10-18 10:35:04 +00:00
|
|
|
@plac.annotations(
|
|
|
|
|
force=("Force overwrite", "flag", "f", bool),
|
2016-11-20 15:48:04 +00:00
|
|
|
data_path=("Path to download model", "option", "d", str)
|
2015-10-18 10:35:04 +00:00
|
|
|
)
|
2016-11-20 15:48:04 +00:00
|
|
|
def main(data_size='all', force=False, data_path=None):
|
2016-10-23 17:46:28 +00:00
|
|
|
if force:
|
|
|
|
|
sputnik.purge(about.__title__, about.__version__)
|
|
|
|
|
|
2016-10-18 22:52:16 +00:00
|
|
|
if data_size in ('all', 'parser'):
|
|
|
|
|
print("Downloading parsing model")
|
2016-11-20 15:48:04 +00:00
|
|
|
download('en', force=False, data_path=data_path)
|
2016-10-18 22:52:16 +00:00
|
|
|
if data_size in ('all', 'glove'):
|
|
|
|
|
print("Downloading GloVe vectors")
|
2016-11-20 15:48:04 +00:00
|
|
|
download('en_glove_cc_300_1m_vectors', force=False, data_path=data_path)
|
2016-01-15 17:01:02 +00:00
|
|
|
|
2015-01-02 10:44:41 +00:00
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
2015-01-31 02:51:56 +00:00
|
|
|
plac.call(main)
|
