Skip to main content

Retrieve pre-trained models and cache it locally

Project description

retrieve

Code Sample

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

retrieve-0.0.8.tar.gz (2.4 kB view details)

Uploaded Source

Built Distribution

retrieve-0.0.8-py3-none-any.whl (3.6 kB view details)

Uploaded Python 3

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

Hashes for retrieve-0.0.8.tar.gz
Algorithm Hash digest
SHA256 36bc84b75910a136e6a3bcb8efd51b4df122f7a8ad5cd0a592d06916b427a549
MD5 c2393d384234303478df556da9a65e7f
BLAKE2b-256 de185cc25f9b612d9975bbf457e34f750fc69d9b5088b48a5927b8a825f6463e

See more details on using hashes here.

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

Hashes for retrieve-0.0.8-py3-none-any.whl
Algorithm Hash digest
SHA256 e958d62b23295e8b1447d19ae8c0fb09ff7b08f04fc37e02a326753c777d2463
MD5 842ada0e798f41b9e44e30201426653a
BLAKE2b-256 fa715f63bf678567339a34369ee3815ff351646801319c26fd0ebfc2c923c124

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page