mirror of https://github.com/explosion/spaCy.git
34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
import sputnik
|
||
|
|
from sputnik.package_list import (PackageNotFoundException,
|
||
|
|
CompatiblePackageNotFoundException)
|
||
|
|
|
||
|
|
from . import about
|
||
|
|
|
||
|
|
|
||
|
|
def download(lang, force=False):
|
||
|
|
if force:
|
||
|
|
sputnik.purge(about.__title__, about.__version__)
|
||
|
|
|
||
|
|
try:
|
||
|
|
sputnik.package(about.__title__, about.__version__, about.__models__[lang]['package'])
|
||
|
|
print("Model already installed. Please run 'python -m "
|
||
|
|
"spacy.%s.download --force' to reinstall." % lang, file=sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
except (PackageNotFoundException, CompatiblePackageNotFoundException):
|
||
|
|
pass
|
||
|
|
|
||
|
|
package = sputnik.install(about.__title__, about.__version__, about.__models__[lang]['package'])
|
||
|
|
|
||
|
|
try:
|
||
|
|
sputnik.package(about.__title__, about.__version__, about.__models__[lang]['package'])
|
||
|
|
except (PackageNotFoundException, CompatiblePackageNotFoundException):
|
||
|
|
print("Model failed to install. Please run 'python -m "
|
||
|
|
"spacy.%s.download --force'." % lang, file=sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
print("Model successfully installed.", file=sys.stderr)
|