Eunchan Lee commited on
Commit
579ed0c
·
1 Parent(s): 256cbd2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def load_zeroshot_classifier():
42
  def get_summarizer(summarizer, sequence:str, maximum_tokens:int, minimum_tokens:int):
43
  output = summarizer(sequence, num_beams=4, max_length=maximum_tokens, min_length=minimum_tokens, do_sample=False)
44
 
45
- metric.add_batch(predictions=output[0].get('summary_text'), references=sequence)
46
  return output[0].get('summary_text')
47
 
48
 
 
42
  def get_summarizer(summarizer, sequence:str, maximum_tokens:int, minimum_tokens:int):
43
  output = summarizer(sequence, num_beams=4, max_length=maximum_tokens, min_length=minimum_tokens, do_sample=False)
44
 
45
+ metric.add(predictions=output[0].get('summary_text'), references=sequence)
46
  return output[0].get('summary_text')
47
 
48