hanch commited on
Commit
045b802
·
1 Parent(s): ac8d16a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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