Kang Suhyun suhyun.kang commited on
Commit
a44d1c3
1 Parent(s): 25a095e

[#36] Update instruction for summarization (#45)

Browse files

This change updates the instruction for summarization more clearly so that it won't change the language of the original text.

Co-authored-by: suhyun.kang <suhyun.kang@yanolja.group>

Files changed (1) hide show
  1. response.py +1 -1
response.py CHANGED
@@ -22,7 +22,7 @@ class Category(enum.Enum):
22
  # TODO(#31): Let the model builders set the instruction.
23
  def get_instruction(category, source_lang, target_lang):
24
  if category == Category.SUMMARIZE.value:
25
- return "Summarize the following text in its original language."
26
  if category == Category.TRANSLATE.value:
27
  return f"Translate the following text from {source_lang} to {target_lang}."
28
 
 
22
  # TODO(#31): Let the model builders set the instruction.
23
  def get_instruction(category, source_lang, target_lang):
24
  if category == Category.SUMMARIZE.value:
25
+ return "Summarize the following text, maintaining the original language of the text in the summary." # pylint: disable=line-too-long
26
  if category == Category.TRANSLATE.value:
27
  return f"Translate the following text from {source_lang} to {target_lang}."
28