added missing generator
Browse files
app.py
CHANGED
|
@@ -184,6 +184,8 @@ print('Done!')
|
|
| 184 |
|
| 185 |
# %% ------------------------------------------- Main App
|
| 186 |
|
|
|
|
|
|
|
| 187 |
pipe = GenerativeQAPipeline(generator, retriever)
|
| 188 |
# pipe = prepare()
|
| 189 |
main(pipe)
|
|
|
|
| 184 |
|
| 185 |
# %% ------------------------------------------- Main App
|
| 186 |
|
| 187 |
+
generator = Seq2SeqGenerator(model_name_or_path="vblagoje/bart_lfqa", use_gpu=False)
|
| 188 |
+
|
| 189 |
pipe = GenerativeQAPipeline(generator, retriever)
|
| 190 |
# pipe = prepare()
|
| 191 |
main(pipe)
|