Spaces:
Runtime error
Runtime error
Update util.py
Browse files
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,
|
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
|