Hamza1702 commited on
Commit
99d2c65
1 Parent(s): be8e82a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,9 +7,10 @@ title = "🤖AI ChatBot"
7
  description = "Building open-domain chatbots is a challenging area for machine learning research."
8
  examples = [["How are you?"]]
9
 
 
10
 
11
- tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
12
- model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
13
 
14
 
15
  def predict(input, history=[]):
 
7
  description = "Building open-domain chatbots is a challenging area for machine learning research."
8
  examples = [["How are you?"]]
9
 
10
+ pipe = pipeline("conversational", model="PygmalionAI/pygmalion-6b")
11
 
12
+ tokenizer = AutoTokenizer.from_pretrained("PygmalionAI/pygmalion-6b")
13
+ model = AutoModelForCausalLM.from_pretrained("PygmalionAI/pygmalion-6b")
14
 
15
 
16
  def predict(input, history=[]):