MartinoMensio / spacy-universal-sentence-encoder

Google USE (Universal Sentence Encoder) for spaCy
MIT License
176 stars 12 forks source link

installation downloads model but to a tmp folder #1

Closed MartinoMensio closed 4 years ago

MartinoMensio commented 4 years ago

This line https://github.com/MartinoMensio/spacy-universal-sentence-encoder-tfhub/blob/e746687dcf084bd5cf7ad0e3012b6388d0fe92c7/setup.py#L2 makes the tfhub download the model to a folder during the installation of the package but then the models folder doesn't get copied when building the wheel. I need to investigate on how a package gets installed. In this way the model would be already downloaded in venv/lib/python3.7/site-packages/universal_sentence_encoder/models and every single run would use the cache.