wiraindrak commited on
Commit
d3f7143
·
1 Parent(s): 34b3fd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -52,13 +52,11 @@ def para_t5(text):
52
  top_k=120,
53
  top_p=0.95,
54
  early_stopping=True,
55
- num_return_sequences=5)
56
- return [
57
- t5_para_tokenizer.decode(
58
- output, skip_special_tokens=True, clean_up_tokenization_spaces=True
59
  )
60
- for output in outputs
61
- ]
62
 
63
  def summarize(text):
64
  t5_ = summ_t5(text)
 
52
  top_k=120,
53
  top_p=0.95,
54
  early_stopping=True,
55
+ num_return_sequences=1)
56
+ return t5_para_tokenizer.decode(
57
+ outputs[0], skip_special_tokens=True, clean_up_tokenization_spaces=True
 
58
  )
59
+
 
60
 
61
  def summarize(text):
62
  t5_ = summ_t5(text)