psyne commited on
Commit
204a311
·
1 Parent(s): 3a5bd74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def pdf_to_text(path, start_page=1, end_page=None):
29
 
30
 
31
  def generate_summary(text, model='bert-base-uncased', ratio=0.2):
32
- model = Summarizer(model_name=model)
33
  summary = model(text, ratio=ratio)
34
  return summary
35
 
 
29
 
30
 
31
  def generate_summary(text, model='bert-base-uncased', ratio=0.2):
32
+ model = Summarizer()
33
  summary = model(text, ratio=ratio)
34
  return summary
35