Spaces:
Sleeping
Sleeping
Update app.py
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("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 |
|
|
|
14 |
@st.cache_resource
|
15 |
def init_models():
|
16 |
#retriever = SentenceTransformer("multi-qa-MiniLM-L6-cos-v1")
|
17 |
+
retriever = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
|
18 |
+
model_name = "sentence-transformers/all-MiniLM-L6-v2"
|
19 |
reader = pipeline(tokenizer=model_name, model=model_name, task='question-answering')
|
20 |
return retriever, reader
|
21 |
|