Views
No views yet
1from transformers import T5ForConditionalGeneration, T5TokenizerFast
2
3tokenizer = T5TokenizerFast.from_pretrained("Annemarie535257/agrifinconnect-chatbot")
4model = T5ForConditionalGeneration.from_pretrained("Annemarie535257/agrifinconnect-chatbot")
5
6inputs = tokenizer("answer the question: How do I apply for a loan?", return_tensors="pt")
7outputs = model.generate(**inputs, max_new_tokens=128)
8print(tokenizer.decode(outputs[0], skip_special_tokens=True))