Francis0917 commited on
Commit
d58ae23
1 Parent(s): 853fdad

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. demo.py +4 -2
demo.py CHANGED
@@ -22,8 +22,10 @@ parser = argparse.ArgumentParser()
22
 
23
  parser.add_argument('--text_input', required=False, type=str, default='g2p_embed')
24
  parser.add_argument('--audio_input', required=False, type=str, default='both')
25
- parser.add_argument('--load_checkpoint_path', required=True, type=str)
26
- parser.add_argument('--keyword_list_length', required=True, type=int)
 
 
27
  parser.add_argument('--stack_extractor', action='store_true')
28
  parser.add_argument('--comment', required=False, type=str)
29
  args = parser.parse_args()
 
22
 
23
  parser.add_argument('--text_input', required=False, type=str, default='g2p_embed')
24
  parser.add_argument('--audio_input', required=False, type=str, default='both')
25
+ parser.add_argument('--load_checkpoint_path', default='/share/nas165/yiting/CL-KWS_202408_v1/checkpoint_results/checkpoint_guided_ctc/20240725-011006')
26
+ parser.add_argument('--keyword_list_length', default=8, type=int)
27
+ # parser.add_argument('--load_checkpoint_path', required=True, type=str)
28
+ # parser.add_argument('--keyword_list_length', required=True, type=int)
29
  parser.add_argument('--stack_extractor', action='store_true')
30
  parser.add_argument('--comment', required=False, type=str)
31
  args = parser.parse_args()