update

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from langchain.llms import HuggingFacePipeline
+from langchain.llms import HuggingFaceHub
 from transformers import AutoTokenizer
 
 # Define the Hugging Face model and tokenizer
@@ -7,7 +7,7 @@ model_name = "microsoft/DialoGPT-medium"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Define the LangChain chat agent
-agent = HuggingFacePipeline(
+agent = HuggingFaceHub(
     model_name=model_name,
     task="text2text-generation",
     tokenizer=tokenizer,