minnehwg commited on
Commit
e0d5b7a
·
verified ·
1 Parent(s): 758e43f

Update util.py

Browse files
Files changed (1) hide show
  1. util.py +2 -2
util.py CHANGED
@@ -127,10 +127,10 @@ def pipeline(url, model, tokenizer):
127
  sub = restore_punctuation(sub)
128
  vie_sub = translate_long(sub)
129
  vie_sub = processed(vie_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
 
127
  sub = restore_punctuation(sub)
128
  vie_sub = translate_long(sub)
129
  vie_sub = processed(vie_sub)
130
+ chunks = split_into_chunks(vie_sub, 700, 2)
131
  sum_para = []
132
  for i in chunks:
133
+ tmp = summarize(i, model, tokenizer, num_beams=5)
134
  sum_para.append(tmp)
135
  sum = ''.join(sum_para)
136
  del sub, vie_sub, sum_para, chunks