ConvoAI is a small conversational AI model trained entirely from scratch using the DailyDialog dataset. It is designed to generate natural and polite responses.
How to use:
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
model = AutoModelForCausalLM.from_pretrained("GBhaveshKumar/ConvoAI")
tokenizer = AutoTokenizer.from_pretrained("GBhaveshKumar/ConvoAI")
chat = pipeline("text-generation", model=model, tokenizer=tokenizer)
response = chat("Hi there! How are you?", max_new_tokens=50)[0]['generated_text']
print(response)
