Timjo88 commited on
Commit
cf4955a
·
1 Parent(s): 993e7c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ import logging
12
 
13
  def start_haystack():
14
  document_store = InMemoryDocumentStore(index="document", embedding_field='embedding', embedding_dim=384, similarity='cosine')
15
- retriever = EmbeddingRetriever(document_store=document_store, embedding_model='sentence-transformers/all-MiniLM-L6-v2', use_gpu=False, top_k=1)
16
  load_data_to_store(document_store,retriever)
17
  pipeline = FAQPipeline(retriever=retriever)
18
  return pipeline
 
12
 
13
  def start_haystack():
14
  document_store = InMemoryDocumentStore(index="document", embedding_field='embedding', embedding_dim=384, similarity='cosine')
15
+ retriever = EmbeddingRetriever(document_store=document_store, embedding_model='sentence-transformers/all-MiniLM-L6-v2', use_gpu=True, top_k=1)
16
  load_data_to_store(document_store,retriever)
17
  pipeline = FAQPipeline(retriever=retriever)
18
  return pipeline