hanch commited on
Commit
3dcc4e8
·
1 Parent(s): 75a550e

changed embeddings

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,8 +14,8 @@ def init_pinecone():
14
  @st.cache_resource
15
  def init_models():
16
  #retriever = SentenceTransformer("multi-qa-MiniLM-L6-cos-v1")
17
- retriever = SentenceTransformer('Xenova/text-embedding-ada-002')
18
- model_name = 'Xenova/text-embedding-ada-002'
19
  reader = pipeline(tokenizer=model_name, model=model_name, task='question-answering')
20
  return retriever, reader
21
 
@@ -58,7 +58,7 @@ def run_query(query):
58
  # force reload
59
  pinecone.init(api_key=PINECONE_KEY, environment=PINE_CONE_ENVIRONMENT)
60
  st.session_state.index = pinecone.Index("dompany-description")
61
- xc = st.session_state.index.query(xq, top_k=3, include_metadata=True)
62
 
63
  results = []
64
  for match in xc['matches']:
 
14
  @st.cache_resource
15
  def init_models():
16
  #retriever = SentenceTransformer("multi-qa-MiniLM-L6-cos-v1")
17
+ retriever = SentenceTransformer("all-MiniLM-L6-v2")
18
+ model_name = "all-MiniLM-L6-v2"
19
  reader = pipeline(tokenizer=model_name, model=model_name, task='question-answering')
20
  return retriever, reader
21
 
 
58
  # force reload
59
  pinecone.init(api_key=PINECONE_KEY, environment=PINE_CONE_ENVIRONMENT)
60
  st.session_state.index = pinecone.Index("dompany-description")
61
+ xc = st.session_state.index.query(xq, top_k=10, include_metadata=True)
62
 
63
  results = []
64
  for match in xc['matches']: