minor fix

arguments.py

@@ -20,7 +20,7 @@ def get_args():
     parser.add_argument(
         '--label_path', type=str, default='./data/esper_demo/labels_all.json', help='style label info file path')
     parser.add_argument(
-        '--checkpoint', type=str, default='./data/esper_demo/ckpt/gpt2_style_2', help='checkpoint file path')
+        '--checkpoint', type=str, default='./data/esper_demo/ckpt/gpt2_style', help='checkpoint file path')
 
     parser.add_argument(
         '--prefix_length', type=int, default=10, help='prefix length for the visual mapper')