minnehwg commited on
Commit
9a3fa77
·
verified ·
1 Parent(s): 3287afb

Update util.py

Browse files
Files changed (1) hide show
  1. util.py +2 -2
util.py CHANGED
@@ -122,7 +122,7 @@ def display(text):
122
  formatted_sentences = [f"• {sentence}" for sentence in unique_sentences]
123
  return formatted_sentences
124
 
125
- def pipeline(url):
126
  trans, sub = get_subtitles(url)
127
  sub = restore_punctuation(sub)
128
  vie_sub = translate_long(sub)
@@ -130,7 +130,7 @@ def pipeline(url):
130
  chunks = split_into_chunks(vie_sub, 700, 3)
131
  sum_para = []
132
  for i in chunks:
133
- tmp = summarize(i, model_aug, tokenizer, num_beams=4)
134
  sum_para.append(tmp)
135
  sum = ''.join(sum_para)
136
  del sub, vie_sub, sum_para, chunks
 
122
  formatted_sentences = [f"• {sentence}" for sentence in unique_sentences]
123
  return formatted_sentences
124
 
125
+ def pipeline(url, model, tokenizer):
126
  trans, sub = get_subtitles(url)
127
  sub = restore_punctuation(sub)
128
  vie_sub = translate_long(sub)
 
130
  chunks = split_into_chunks(vie_sub, 700, 3)
131
  sum_para = []
132
  for i in chunks:
133
+ tmp = summarize(i, model, tokenizer, num_beams=4)
134
  sum_para.append(tmp)
135
  sum = ''.join(sum_para)
136
  del sub, vie_sub, sum_para, chunks