Y5Bot / README.md
jojo-ai-mst's picture
Create README.md
58b2f0e verified
|
raw
history blame
No virus
844 Bytes
metadata
license: mit
language:
  - en
library_name: transformers
tags:
  - Question Answering
datasets:
  - yahoo_answers_qa
metrics:
  - rouge

Y5 Bot

Y5 Bot is a text2text transfer transformer model woking in English Language.

It is lightweight, fast, and easy to use for text generation.

How to use

pip install transformers
model = T5ForConditionalGeneration.from_pretrained("jojo-ai-mst/Y5Bot")
tokenizer = T5Tokenizer.from_pretrained("jojo-ai-mst/Y5Bot")

my_question = "What do you think about the benefit of Artificial Intelligence?"
inputs = "Please answer to this question: " + my_question

inputs = tokenizer(inputs, return_tensors="pt")
outputs = model.generate(**inputs)
answer = tokenizer.decode(outputs[0])

print(answer)

References