fix instruction

tasks/summarization.py

@@ -97,7 +97,7 @@ def _translate_example(example: Dict[str, str], src_language: str, target_langua
         print(e)
 
 
-def create_instruction(lang: str, expected_output: str):
+def create_instruction(lang: str, instruction_language: str, expected_output: str):
     basic_instruction = (
         f"Write a summary of the given <Text> \n The output should be in {expected_output} "
         f"\n The output must be up to 2 sentences maximum!!!"
@@ -105,7 +105,7 @@ def create_instruction(lang: str, expected_output: str):
     print(lang)
     return (
         basic_instruction
-        if expected_output == "english"
+        if instruction_language == "english"
         else _translate_instruction(basic_instruction, target_language=lang)
     )
 
@@ -127,7 +127,7 @@ def construct_prompt(
 ):
     if not instruction:
         print(lang)
-        instruction = create_instruction(lang, config['prefix'])
+        instruction = create_instruction(lang, config['prefix'], config['output'])
 
     example_prompt = PromptTemplate(
         input_variables=["summary", "text"], template="Text: {text}\nSummary: {summary}"