dyxohjl666 commited on
Commit
2a0f616
1 Parent(s): ac2c93f

Fix a bug in summarization

Browse files
Files changed (1) hide show
  1. summarization.py +1 -1
summarization.py CHANGED
@@ -24,7 +24,7 @@ def ssum_for_file(input, num_beams=1, num_return_sequences=1) -> List[Tuple[str,
24
  results = ssum_pipeline.predict(filename, type="txt", num_beams=num_beams,
25
  num_return_sequences=num_return_sequences, save_results=False)
26
  elif filename[-4:] == ".pdf":
27
- results = summ_pipeline.predict(filename, num_beams=num_beams, num_return_sequences=num_return_sequences, save_results=False)
28
  else:
29
  return [("File Format Error !", None)]
30
 
 
24
  results = ssum_pipeline.predict(filename, type="txt", num_beams=num_beams,
25
  num_return_sequences=num_return_sequences, save_results=False)
26
  elif filename[-4:] == ".pdf":
27
+ results = ssum_pipeline.predict(filename, num_beams=num_beams, num_return_sequences=num_return_sequences, save_results=False)
28
  else:
29
  return [("File Format Error !", None)]
30