BentoML artifact framework for TensorFlow Hub models
Project description
tfhubartifact
BentoML artifact framework for TensorFlow Hub models.
Installation:
pip install tfhubartifact==0.0.1
Usage example (decorate service):
from tfhubartifact.TensorFlowHubModelArtifact import TensorFlowHubModelArtifact
@artifacts([TensorFlowHubModelArtifact('embedder')])
class MyBentoService(BentoService):
Usage example (package model):
svc = MyBentoService()
svc.pack('embedder', my_embedding_model_path)
Alternatively, during training:
svc.pack('embedder', {'embedder': my_embedder})
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
tfhubartifact-0.0.4.tar.gz
(2.4 kB
view hashes)
Built Distribution
Close
Hashes for tfhubartifact-0.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5b6bea5dad469f1e5b5ccd989ee1e1e98f0f28e26b77459a5849ede00a8b48bb |
|
MD5 | 48456e1590dc52a44505875c55bcaee5 |
|
BLAKE2b-256 | 2a4ea28f0a6a9abe4fd3bafa45973a65dba2574b9118470ca0ae71c0d376ec89 |