关键词生成文本

该项目旨在构建一个能够接收关键词作为输入并生成句子作为输出的模型。

模型说明：

目前已构建两个模型：

• 基于T5-base架构的模型（大小850MB），访问地址：https://huggingface.co/gagan3012/keytotext
• 基于T5-small架构的模型（大小230MB），访问地址：https://huggingface.co/gagan3012/keytotext-small

使用方法：

from transformers import AutoTokenizer, AutoModelWithLMHead

tokenizer = AutoTokenizer.from_pretrained("gagan3012/keytotext-small")

model = AutoModelWithLMHead.from_pretrained("gagan3012/keytotext-small")

在线演示：

演示地址：https://share.streamlit.io/gagan3012/keytotext/app.py

生成示例：

输入关键词['印度', '婚礼'] → 输出句子："我们正在新德里举办三场结婚周年庆典来庆祝今天这个日子。"