Alberto Carmona commited on
Commit
af90128
·
1 Parent(s): 47d82ab

Remove the instruction part of the output

Browse files
Files changed (1) hide show
  1. functions.py +1 -0
functions.py CHANGED
@@ -36,6 +36,7 @@ def summarize_text(text: str):
36
  generation_config=generation_config
37
  )
38
  output = tokenizer.decode(output_tokens[0], skip_special_tokens=True)
 
39
  print(['summarize_text', 'end'])
40
  return output
41
 
 
36
  generation_config=generation_config
37
  )
38
  output = tokenizer.decode(output_tokens[0], skip_special_tokens=True)
39
+ output = output.replace(input_text, '')
40
  print(['summarize_text', 'end'])
41
  return output
42