Views
No views yet
google/flan-t5-large specifically tailored to act as a mental health support assistant trained on the MentalChat-16k dataset.google/flan-t5-largetext2text-generation1from transformers import pipeline
2
3chatbot = pipeline("text2text-generation", model="sarthak0226/flan-t5-large-mentalchat")
4
5prompt = "I'm having trouble sleeping lately. What can I do?"
6response = chatbot(prompt, max_length=100, do_sample=True, top_k=50, top_p=0.95)
7
8print(response[0]["generated_text"])