Blaise-g commited on
Commit
afcaa5c
Β·
1 Parent(s): 45441d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -87,6 +87,13 @@ def proc_submission(
87
  batch_length=token_batch_length,
88
  **settings,
89
  )
 
 
 
 
 
 
 
90
  sum_text = [f"Section {i}: " + s["summary"][0] for i, s in enumerate(_summaries)]
91
  compression_rate = [
92
  f" - Section {i}: {round(s['compression_rate'],3)}"
 
87
  batch_length=token_batch_length,
88
  **settings,
89
  )
90
+ _summaries = summarize_via_tokenbatches(
91
+ tr_in,
92
+ model_tldr if (summary_type == "TLDR") else model_det,
93
+ tokenizer_tldr if (summary_type == "TLDR") else tokenizer_det,
94
+ batch_length=token_batch_length,
95
+ **settings,
96
+ )
97
  sum_text = [f"Section {i}: " + s["summary"][0] for i, s in enumerate(_summaries)]
98
  compression_rate = [
99
  f" - Section {i}: {round(s['compression_rate'],3)}"