notSoNLPnerd commited on
Commit
dbf4cbb
·
1 Parent(s): 34f007a

update pipeline yaml

Browse files
Files changed (1) hide show
  1. retriever_reader.yml +33 -0
retriever_reader.yml ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ components:
2
+ - name: DocumentStore
3
+ type: InMemoryDocumentStore
4
+ - name: tfidf_ret
5
+ params:
6
+ document_store: DocumentStore
7
+ type: TfidfRetriever
8
+ - name: reader
9
+ params:
10
+ # set model_name_or_path HF model_name
11
+ model_name_or_path: deepset/minilm-uncased-squad2
12
+ # here we have used a base-cpu image
13
+ use_gpu: false
14
+ type: FARMReader
15
+ - name: TextFileConverter
16
+ type: TextConverter
17
+ pipelines:
18
+ - name: query
19
+ nodes:
20
+ - inputs: [Query]
21
+ name: tfidf_ret
22
+ - inputs: [tfidf_ret]
23
+ name: reader
24
+ - name: indexing
25
+ nodes:
26
+ - name: TextFileConverter
27
+ inputs: [ File ]
28
+ - name: tfidf_ret
29
+ inputs: [ TextFileConverter ]
30
+ - name: DocumentStore
31
+ inputs: [ tfidf_ret ]
32
+ version: ignore
33
+