anakin87 commited on
Commit
53cefc5
·
1 Parent(s): b175f36

retry to fix

Browse files
Files changed (1) hide show
  1. app_utils/backend_utils.py +2 -1
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, task_name="text2text-generation")
 
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