Update app.py

app.py

@@ -8,11 +8,10 @@ title = "🤖AI ChatBot"
 description = "Building open-domain chatbots is a challenging area for machine learning research."
 examples = [["How are you?"]]
 
-pipe = pipeline("conversational", model="PygmalionAI/pygmalion-6b")
-
-tokenizer = AutoTokenizer.from_pretrained("PygmalionAI/pygmalion-6b")
-model = AutoModelForCausalLM.from_pretrained("PygmalionAI/pygmalion-6b")
+pipe = pipeline("conversational", model="microsoft/DialoGPT-medium")
 
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
 def predict(input, history=[]):
     # tokenize the new input sentence