Skip to content

Text completion with Hugging Face and TensorFlow.js running on Node.js

License

Notifications You must be signed in to change notification settings

katanaml/text-completion

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Katana ML Text Completion 🤗

Description

Runs with with Hugging Face DistilBERT and TensorFlow.js on Node.js

  • distilbert-model - converter from Hugging Face model format to TensorFlow SavedModel

This is work in progress repo. I'm working on TensorFlow.js part to load SavedModel and run inference.

References

Author

Katana ML, Andrej Baranovskij

License

Licensed under the Apache License, Version 2.0. Copyright 2020-2021 Katana ML, Andrej Baranovskij. Copy of the license.