Update app.py

app.py

@@ -4,8 +4,8 @@ from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
 import pickle
-from langchain.llms import HuggingFaceEndpoint
-from langchain.embeddings import HuggingFaceEmbeddings
+from langchain_community.llms import HuggingFaceEndpoint
+from langchain_community.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",
-    api_key=os.getenv('HUGGINGFACEHUB_API_TOKEN')
+    model_kwargs={"api_key": os.getenv('HUGGINGFACEHUB_API_TOKEN')}
 )
 
 # Initialize the HuggingFace embeddings
@@ -112,7 +112,7 @@ def query_text(text):
 def main():
     upload_interface = gr.Interface(
         fn=upload_files,
-        inputs=gr.inputs.File(file_count="multiple", label="Upload DOCX files"),
+        inputs=gr.File(file_count="multiple", label="Upload DOCX files"),
         outputs="text",
         title="Upload DOCX Files",
         description="Upload DOCX files to process and add to the FAISS index."
@@ -143,3 +143,5 @@ if __name__ == "__main__":
 
 
 
+
+