Tuana commited on
Commit
abe071a
·
1 Parent(s): 8598810

switching to roberta-base-sqaud2-distilled

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def start_haystack():
15
  document_store = InMemoryDocumentStore()
16
  load_and_write_data(document_store)
17
  retriever = TfidfRetriever(document_store=document_store)
18
- reader = FARMReader(model_name_or_path="deepset/tinyroberta-squad2", use_gpu=True)
19
  pipeline = ExtractiveQAPipeline(reader, retriever)
20
  return pipeline
21
 
 
15
  document_store = InMemoryDocumentStore()
16
  load_and_write_data(document_store)
17
  retriever = TfidfRetriever(document_store=document_store)
18
+ reader = FARMReader(model_name_or_path="deepset/roberta-base-squad2-distilled", use_gpu=True)
19
  pipeline = ExtractiveQAPipeline(reader, retriever)
20
  return pipeline
21