Retrieve pre-trained models and cache it locally
Project description
retrieve
No-frills library to download pre-trained models, cache it and return the local path.
Installation
pip install -U retrieve
Motivation
When using pre-trained ML models in your projects, majority of them require you to manually download the data/model weights and then specify the path in your code. Everyone in your team has to go through the same trouble of manually setting this up before they can run your models.
The idea with this library is automate this and make using pre-trained models as easy as possible.
Usage
Pass the pre-trained model path to the retrieve.url(...) method and it will return you a filepath to the file. If the file is not already download, it is fetched and shows a progress of download.
import retrieve
# URL for the pre-trained model
pretrained_model_url = '...'
# Get local path to the pre-trained model
path = retrieve.url(pretrained_model_url)
If source is a zip file, it's automatically extracted out and the extracted folder path is returned.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file retrieve-0.0.8.tar.gz
.
File metadata
- Download URL: retrieve-0.0.8.tar.gz
- Upload date:
- Size: 2.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.1 importlib_metadata/4.5.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.61.0 CPython/3.9.5
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 36bc84b75910a136e6a3bcb8efd51b4df122f7a8ad5cd0a592d06916b427a549 |
|
MD5 | c2393d384234303478df556da9a65e7f |
|
BLAKE2b-256 | de185cc25f9b612d9975bbf457e34f750fc69d9b5088b48a5927b8a825f6463e |
File details
Details for the file retrieve-0.0.8-py3-none-any.whl
.
File metadata
- Download URL: retrieve-0.0.8-py3-none-any.whl
- Upload date:
- Size: 3.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.1 importlib_metadata/4.5.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.61.0 CPython/3.9.5
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | e958d62b23295e8b1447d19ae8c0fb09ff7b08f04fc37e02a326753c777d2463 |
|
MD5 | 842ada0e798f41b9e44e30201426653a |
|
BLAKE2b-256 | fa715f63bf678567339a34369ee3815ff351646801319c26fd0ebfc2c923c124 |