Update app.py

app.py

@@ -4,8 +4,8 @@ from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
 import pickle
-from langchain_huggingface.llms import HuggingFaceEndpoint
-from langchain_huggingface.embeddings import HuggingFaceEmbeddings
+from langchain.llms import HuggingFaceEndpoint
+from langchain.embeddings import HuggingFaceEmbeddings
 import gradio as gr
 
 # Initialize the embedding model
@@ -14,7 +14,7 @@ embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
 # Initialize the HuggingFace LLM
 llm = HuggingFaceEndpoint(
     endpoint_url="https://api-inference.huggingface.co/models/gpt2",
-    model_kwargs={"api_key": os.getenv('HUGGINGFACEHUB_API_TOKEN')}
+    api_key=os.getenv('HUGGINGFACEHUB_API_TOKEN')
 )
 
 # Initialize the HuggingFace embeddings
@@ -142,3 +142,4 @@ if __name__ == "__main__":
 
 
 
+