🚀 Socket Launch Week 🚀 Day 4: Introducing Historical Analytics.Learn More
Socket
Sign inDemoInstall
Socket

retrieve

Package Overview
Dependencies
Maintainers
1
Alerts
File Explorer

Advanced tools

Socket logo

Install Socket

Detect and block malicious and high-risk dependencies

Install

retrieve

Retrieve pre-trained models and cache it locally

0.0.8
PyPI
Maintainers
1

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.

FAQs

Did you know?

Socket

Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.

Install

Related posts