wing-nus commited on
Commit
96bfa0d
1 Parent(s): b6d6887

Update controlled_summarization.py

Browse files
Files changed (1) hide show
  1. controlled_summarization.py +1 -1
controlled_summarization.py CHANGED
@@ -128,7 +128,7 @@ def ctrlsum_for_file(input=None, length=None, keywords="", text="", url="") -> L
128
  keywords = None
129
  if l == "":
130
  l = None
131
- return ctrlsum_for_str(raw_text, l, keywords), raw_text, None
132
 
133
  filename = download_pdf(url, './cache/')
134
  else:
 
128
  keywords = None
129
  if l == "":
130
  l = None
131
+ return ctrlsum_for_str(raw_text, int(l), keywords), raw_text, None
132
 
133
  filename = download_pdf(url, './cache/')
134
  else: