codelion commited on
Commit
9471545
·
verified ·
1 Parent(s): 2098b64

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -4
main.py CHANGED
@@ -4,7 +4,7 @@ import streamlit as st
4
  import anthropic
5
 
6
  # from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
7
- from langchain_community.embeddings import HuggingFaceInstructEmbeddings
8
  from langchain_community.vectorstores import SupabaseVectorStore
9
  from langchain_community.llms import HuggingFaceEndpoint
10
  from langchain_community.vectorstores import SupabaseVectorStore
@@ -26,12 +26,10 @@ username = st.secrets.username
26
  supabase: Client = create_client(supabase_url, supabase_key)
27
  logger = get_logger(__name__)
28
 
29
-
30
-
31
  model_name = "BAAI/bge-large-en-v1.5"
32
  model_kwargs = {'device': 'cpu'}
33
  encode_kwargs = {'normalize_embeddings': True}
34
- embeddings = HuggingFaceInstructEmbeddings(
35
  model_name=model_name,
36
  model_kwargs=model_kwargs,
37
  encode_kwargs=encode_kwargs
 
4
  import anthropic
5
 
6
  # from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
7
+ from langchain_community.embeddings import HuggingFaceBgeEmbeddings
8
  from langchain_community.vectorstores import SupabaseVectorStore
9
  from langchain_community.llms import HuggingFaceEndpoint
10
  from langchain_community.vectorstores import SupabaseVectorStore
 
26
  supabase: Client = create_client(supabase_url, supabase_key)
27
  logger = get_logger(__name__)
28
 
 
 
29
  model_name = "BAAI/bge-large-en-v1.5"
30
  model_kwargs = {'device': 'cpu'}
31
  encode_kwargs = {'normalize_embeddings': True}
32
+ embeddings = HuggingFaceBgeEmbeddings(
33
  model_name=model_name,
34
  model_kwargs=model_kwargs,
35
  encode_kwargs=encode_kwargs