Spaces:

anakin87
/

fact-checking-rocks

Running

anakin87 commited on Aug 8

Commit

b175f36

1 Parent(s): ad6e7a3

specify task_name

Files changed (1) hide show

app_utils/backend_utils.py CHANGED Viewed

@@ -55,7 +55,7 @@ def start_haystack():
     pipe.add_node(component=retriever, name="retriever", inputs=["Query"])
     pipe.add_node(component=entailment_checker, name="ec", inputs=["retriever"])
-    prompt_node = PromptNode(model_name_or_path=PROMPT_MODEL, max_length=150)
     return pipe, prompt_node

     pipe.add_node(component=retriever, name="retriever", inputs=["Query"])
     pipe.add_node(component=entailment_checker, name="ec", inputs=["retriever"])
+    prompt_node = PromptNode(model_name_or_path=PROMPT_MODEL, max_length=150, task_name="text2text-generation")
     return pipe, prompt_node