A fine-tuned conversational AI model designed to be supportive, caring, and romantic in conversations.
Mr.Bully is a fine-tuned version of DialoGPT-small, specifically trained to provide supportive and romantic responses in conversations. The model maintains a caring, encouraging personality while engaging users in meaningful dialogue.
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("sambitsingha/mrbully-chatbot")
4model = AutoModelForCausalLM.from_pretrained("sambitsingha/mrbully-chatbot")
5
6# Format your input
7prompt = "User: Hello!\nMr.Bully:"
8inputs = tokenizer(prompt, return_tensors="pt")
9
10# Generate response
11with torch.no_grad():
12 outputs = model.generate(
13 inputs.input_ids,
14 max_length=100,
15 temperature=0.7,
16 do_sample=True,
17 pad_token_id=tokenizer.eos_token_id
18 )
19
20response = tokenizer.decode(outputs[0], skip_special_tokens=True)
21print(response)
This model was fine-tuned on conversational data with a focus on supportive and romantic dialogue patterns.
