sambitsingha commited on
Commit
10cc4b3
·
verified ·
1 Parent(s): 7e2abc4

Add model card

Browse files
Files changed (1) hide show
  1. README.md +36 -0
README.md ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: mit
4
+ tags:
5
+ - conversational
6
+ - chatbot
7
+ - fine-tuned
8
+ datasets:
9
+ - custom
10
+ ---
11
+
12
+ # Mr.Bully Chatbot Model
13
+
14
+ This is a fine-tuned conversational AI model trained on custom dialogue data.
15
+
16
+ ## Usage
17
+
18
+ ```python
19
+ from transformers import AutoTokenizer, AutoModelForCausalLM
20
+
21
+ tokenizer = AutoTokenizer.from_pretrained("sambitsingha/mrbully-chatbot")
22
+ model = AutoModelForCausalLM.from_pretrained("sambitsingha/mrbully-chatbot")
23
+
24
+ # Generate response
25
+ prompt = "User: Hello\nMr.Bully:"
26
+ inputs = tokenizer.encode(prompt, return_tensors="pt")
27
+ outputs = model.generate(inputs, max_length=150, temperature=0.7)
28
+ response = tokenizer.decode(outputs[0], skip_special_tokens=True)
29
+ print(response)
30
+ ```
31
+
32
+ ## Model Details
33
+
34
+ - Base model: GPT-2 (or specify your base model)
35
+ - Fine-tuned on custom conversational data
36
+ - Optimized for romantic/affectionate conversations