Deepakraj2006 commited on
Commit
50e80a6
·
verified ·
1 Parent(s): ba51932

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=130, 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)
 
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)