Spaces:
Sleeping
Sleeping
changed embeddings
Browse files
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(
|
18 |
-
model_name =
|
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=
|
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']:
|