Update app.py

app.py

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