nik-one commited on
Commit
041e3bb
·
verified ·
1 Parent(s): 0458219

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ if "memory" not in st.session_state:
38
  @st.cache_resource
39
  def load_embeddings():
40
  """Load and cache the embeddings model."""
41
- return HuggingFaceEmbeddings(model_name="nlpaueb/legal-bert-base-uncased")
42
 
43
  embeddings = load_embeddings()
44
  db = FAISS.load_local("ipc_embed_db", embeddings, allow_dangerous_deserialization=True)
 
38
  @st.cache_resource
39
  def load_embeddings():
40
  """Load and cache the embeddings model."""
41
+ return HuggingFaceEmbeddings(model_name="law-ai/InLegalBERT")
42
 
43
  embeddings = load_embeddings()
44
  db = FAISS.load_local("ipc_embed_db", embeddings, allow_dangerous_deserialization=True)