按键文字
想法是建立一个将关键字作为输入并生成句子作为输出的模型。
模型:
已经建立了两个模型:
使用T5的基本大小= 850 MB可以在这里找到: :
使用T5,小尺寸= 230 MB,可以在这里找到: :
用法:
from transformers import AutoTokenizer , AutoModelWithLMHead
tokenizer = AutoTokenizer . from_pretrained ( "gagan3012/keytotext-small" )
model = AutoModelWithLMHead . from_pretrained ( "gagan3012/keytotext-small" )
这使用了我构建的自定义流式compopnet:
pip install streamlit-tags
也可以在上找到安装
演示:
例
1