Update App.py

app.py

@@ -1,15 +1,20 @@
+
+from transformers import pipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 import torch
 
+# Use a pipeline as a high-level helper
+
+pipe = pipeline("conversational", model="PygmalionAI/pygmalion-6b")
 
 title = "🤖AI ChatBot"
 description = "Building open-domain chatbots is a challenging area for machine learning research."
 examples = [["How are you?"]]
 
 
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+tokenizer = AutoTokenizer.from_pretrained("PygmalionAI/pygmalion-6b")
+model = AutoModelForCausalLM.from_pretrained("PygmalionAI/pygmalion-6b")
 
 
 def predict(input, history=[]):