Spaces:
Running
Running
switching to roberta-base-sqaud2-distilled
Browse files
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/
|
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 |
|