Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 from langchain_community.llms import HuggingFaceEndpoint
 from langchain.prompts import PromptTemplate
 from transformers import pipeline
+from langchain_huggingface import HuggingFaceEndpoint
 # Initialize the chatbot
 HF_TOKEN = os.getenv("HF_TOKEN")
 # Use a pipeline as a high-level helper