update

app.py

@@ -1,17 +1,11 @@
 import gradio as gr
-from langchain.llms.huggingface_hub import HuggingFaceHub
+from langchain.chains.conversation import ConversationChain
 from transformers import AutoTokenizer
 
-# Define the Hugging Face model and tokenizer
+# Define the LangChain chat agent
 model_name = "microsoft/DialoGPT-medium"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-
-# Define the LangChain chat agent
-agent = HuggingFaceHub(
-    model_name=model_name,
-    task="text2text-generation",
-    tokenizer=tokenizer,
-)
+agent = ConversationChain(llm=model_name, tokenizer=tokenizer)
 
 # Define the Gradio interface
 def chatbot_interface(input_text):
@@ -23,8 +17,8 @@ gradio_app = gr.Interface(
     fn=chatbot_interface,
     inputs=gr.inputs.Textbox(prompt="Say something..."),
     outputs=gr.outputs.Textbox(),
-    title="LangChain Chatbot",
-    description="A chatbot interface powered by LangChain and Hugging Face.",
+    title="ConversationChain Chatbot",
+    description="A chatbot interface powered by ConversationChain and Hugging Face.",
 )
 
 # Run the Gradio app