Update app.py

app.py

@@ -9,7 +9,7 @@ subprocess.check_call([
     "huggingface_hub",
     "faiss-cpu",
     "langchain-together",
-    "sentence-transformers"
+    "transformers"
 ])
 
 import time
@@ -21,6 +21,7 @@ from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferWindowMemory
 from langchain.chains import ConversationalRetrievalChain
 from langchain_together import Together
+from transformers import AutoTokenizer, AutoModel
 
 # Footer code
 def footer():
@@ -64,7 +65,9 @@ if "memory" not in st.session_state:
 @st.cache_resource
 def load_embeddings():
     """Load and cache the embeddings model."""
-    return HuggingFaceEmbeddings(model_name="law-ai/InLegalBERT")
+    tokenizer = AutoTokenizer.from_pretrained("nlpaueb/legal-bert")
+    model = AutoModel.from_pretrained("nlpaueb/legal-bert")
+    return HuggingFaceEmbeddings(model=model, tokenizer=tokenizer)
 
 embeddings = load_embeddings()
 db = FAISS.load_local("ipc_embed_db", embeddings, allow_dangerous_deserialization=True)