DD0101 commited on
Commit
525d89c
1 Parent(s): aa383b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ parser = argparse.ArgumentParser()
60
 
61
  # parser.add_argument("--input_file", default="sample_pred_in.txt", type=str, help="Input file for prediction")
62
  # parser.add_argument("--output_file", default="sample_pred_out.txt", type=str, help="Output file for prediction")
63
- parser.add_argument("--model_dir", default="Customized_IDSF/JointBERT-CRF_PhoBERTencoder", type=str, help="Path to save, load model")
64
 
65
  parser.add_argument("--batch_size", default=32, type=int, help="Batch size for prediction")
66
  parser.add_argument("--no_cuda", action="store_true", help="Avoid using CUDA when available")
 
60
 
61
  # parser.add_argument("--input_file", default="sample_pred_in.txt", type=str, help="Input file for prediction")
62
  # parser.add_argument("--output_file", default="sample_pred_out.txt", type=str, help="Output file for prediction")
63
+ parser.add_argument("--model_dir", default="/home/user/app/Customized_IDSF/JointBERT-CRF_PhoBERTencoder", type=str, help="Path to save, load model")
64
 
65
  parser.add_argument("--batch_size", default=32, type=int, help="Batch size for prediction")
66
  parser.add_argument("--no_cuda", action="store_true", help="Avoid using CUDA when available")