Spaces:
Running
Running
retry to fix
Browse files
app_utils/backend_utils.py
CHANGED
@@ -55,7 +55,8 @@ def start_haystack():
|
|
55 |
pipe.add_node(component=retriever, name="retriever", inputs=["Query"])
|
56 |
pipe.add_node(component=entailment_checker, name="ec", inputs=["retriever"])
|
57 |
|
58 |
-
prompt_node = PromptNode(model_name_or_path=PROMPT_MODEL, max_length=150,
|
|
|
59 |
|
60 |
return pipe, prompt_node
|
61 |
|
|
|
55 |
pipe.add_node(component=retriever, name="retriever", inputs=["Query"])
|
56 |
pipe.add_node(component=entailment_checker, name="ec", inputs=["retriever"])
|
57 |
|
58 |
+
prompt_node = PromptNode(model_name_or_path=PROMPT_MODEL, max_length=150,
|
59 |
+
model_kwargs={"task_name": "text2text-generation"})
|
60 |
|
61 |
return pipe, prompt_node
|
62 |
|