Update app.py

app.py

@@ -1,20 +1,15 @@
-
-from transformers import pipeline
-from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 import torch
 
-# Use a pipeline as a high-level helper
-
-pipe = pipeline("conversational", model="facebook/blenderbot-400M-distill")
 
 title = "🤖AI ChatBot"
 description = "Building open-domain chatbots is a challenging area for machine learning research."
 examples = [["How are you?"]]
 
 
-tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
-model = AutoModelForSeq2SeqLM.from_pretrained("facebook/blenderbot-400M-distill")
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 
 
 def predict(input, history=[]):
@@ -49,4 +44,4 @@ gr.Interface(
     inputs=["text", "state"],
     outputs=["chatbot", "state"],
     theme="finlaymacklon/boxy_violet",
-).launch(share=True)
+).launch()