Hugging Face's logo

Spaces:
Tuana
/
GoT-QA-Haystack
Running

App Files Community
Tuana commited on
Commit
3710fa9
·
1 Parent(s): 4ef8a52

Change to distilled model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import json
9
  #Haystack Components
10
  document_store = InMemoryDocumentStore()
11
  retriever = TfidfRetriever(document_store=document_store)
12
- reader = FARMReader(model_name_or_path="deepset/roberta-base-squad2", use_gpu=True)
13
  pipeline = ExtractiveQAPipeline(reader, retriever)
14
 
15
  @st.cache
 
9
  #Haystack Components
10
  document_store = InMemoryDocumentStore()
11
  retriever = TfidfRetriever(document_store=document_store)
12
+ reader = FARMReader(model_name_or_path="deepset/tinyroberta-squad2", use_gpu=True)
13
  pipeline = ExtractiveQAPipeline(reader, retriever)
14
 
15
  @st.cache