Update util.py

util.py

@@ -122,7 +122,7 @@ def display(text):
     formatted_sentences = [f"• {sentence}" for sentence in unique_sentences]
     return formatted_sentences
 
-def pipeline(url):
+def pipeline(url, model, tokenizer):
     trans, sub = get_subtitles(url)
     sub = restore_punctuation(sub)
     vie_sub = translate_long(sub)
@@ -130,7 +130,7 @@ def pipeline(url):
     chunks = split_into_chunks(vie_sub, 700, 3)
     sum_para = []
     for i in chunks:
-        tmp = summarize(i, model_aug, tokenizer, num_beams=4)
+        tmp = summarize(i, model, tokenizer, num_beams=4)
         sum_para.append(tmp)
     sum = ''.join(sum_para)
     del sub, vie_sub, sum_para, chunks