Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -155,7 +155,7 @@ def process_company(company):
|
|
155 |
for i, chunk in enumerate(chunks):
|
156 |
print(f"Summarizing chunk {i+1}/{len(chunks)}...")
|
157 |
inputs = bart_tokenizer([chunk], max_length=1024, return_tensors='pt', truncation=True)
|
158 |
-
summary_ids = bart_model.generate(inputs.input_ids, num_beams=2, max_length=
|
159 |
chunk_summary = bart_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
160 |
chunk_summaries.append(chunk_summary)
|
161 |
final_summary = ' '.join(chunk_summaries)
|
|
|
155 |
for i, chunk in enumerate(chunks):
|
156 |
print(f"Summarizing chunk {i+1}/{len(chunks)}...")
|
157 |
inputs = bart_tokenizer([chunk], max_length=1024, return_tensors='pt', truncation=True)
|
158 |
+
summary_ids = bart_model.generate(inputs.input_ids, num_beams=2, max_length=50, min_length=30, early_stopping=True)
|
159 |
chunk_summary = bart_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
160 |
chunk_summaries.append(chunk_summary)
|
161 |
final_summary = ' '.join(chunk_summaries)
|