update script

src/test.py → src/main.py

@@ -18,13 +18,13 @@ def main():
  parser.add_argument('-m',
  '--mode',
  type=str,
- help='the mode of summarization\n -pegasus\n -nltk\n',
- required=False,
- default="ntlk")
+ help='the mode of summarization\n -pegasus\n -nltk\n -sumy\n',
+ required=True,
+ default="nltk")
 
  args = parser.parse_args()
 
- if args.mode == "ntlk":
+ if args.mode == "nltk":
  summarizer_NTLK = nltk_summarizer.PDFSummarizer(args.file)
  summarizer_NTLK.summarize()
  summarizer_NTLK.export(args.output)

test.sh

@@ -0,0 +1,10 @@
+#!/bin/bash
+
+file="YouTubers_Not_madeForKids_Detecting_Channels_Sharing_Inappropriate_Videos_Targeting_Children"
+mkdir -p data/output
+echo -e "Testing summarization methods ...\n"
+echo -e "\nSummarization nltk:\n"
+time python src/main.py -f data/input/${file}.pdf -o data/output/${file}_nltk.txt -m nltk
+# time python src/main.py -f data/input/${file}.pdf -o data/output/${file}_pegasus.txt -m pegasus
+echo -e "\nSummarization sumy:\n"
+time python src/main.py -f data/input/${file}.pdf -o data/output/${file}_summy.txt -m sumy