gagan3012 commited on
Commit
cb54dbf
·
1 Parent(s): ec3086f

Update cli.py

Browse files
Files changed (1) hide show
  1. t5s/cli.py +1 -0
t5s/cli.py CHANGED
@@ -3,6 +3,7 @@ import os
3
  import subprocess
4
  import sys
5
 
 
6
  arg_parser = argparse.ArgumentParser(
7
  description="T5 Summarisation Using Pytorch Lightning", prog="t5s"
8
  )
 
3
  import subprocess
4
  import sys
5
 
6
+ import yaml
7
  arg_parser = argparse.ArgumentParser(
8
  description="T5 Summarisation Using Pytorch Lightning", prog="t5s"
9
  )