Update app.py
Browse files
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)}"
|