Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 from langchain.agents import Agent
-from langchain_community.llms import HuggingFaceEndpoint
+from langchain_huggingface.llms import HuggingFaceEndpoint
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
 
@@ -10,7 +10,7 @@ HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN', 'your_huggingfa
 
 # Initialize the LLM from Hugging Face Hub
 llm = HuggingFaceEndpoint(endpoint_url="https://api-inference.huggingface.co/models/gpt2", 
-                          headers={"Authorization": f"Bearer {HUGGINGFACEHUB_API_TOKEN}"})
+                          model_kwargs={"headers": {"Authorization": f"Bearer {HUGGINGFACEHUB_API_TOKEN}"}})
 
 # Initialize embeddings
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")