DD0101 commited on
Commit
a15ee24
1 Parent(s): f353d5a

change default of --model_dir

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -213,7 +213,7 @@ if __name__ == "__main__":
213
 
214
  # parser.add_argument("--input_file", default="sample_pred_in.txt", type=str, help="Input file for prediction")
215
  # parser.add_argument("--output_file", default="sample_pred_out.txt", type=str, help="Output file for prediction")
216
- parser.add_argument("--model_dir", default="./atis_model", type=str, help="Path to save, load model")
217
 
218
  parser.add_argument("--batch_size", default=32, type=int, help="Batch size for prediction")
219
  parser.add_argument("--no_cuda", action="store_true", help="Avoid using CUDA when available")
 
213
 
214
  # parser.add_argument("--input_file", default="sample_pred_in.txt", type=str, help="Input file for prediction")
215
  # parser.add_argument("--output_file", default="sample_pred_out.txt", type=str, help="Output file for prediction")
216
+ parser.add_argument("--model_dir", default="./JointBERT-CRF_PhoBERTencoder", type=str, help="Path to save, load model")
217
 
218
  parser.add_argument("--batch_size", default=32, type=int, help="Batch size for prediction")
219
  parser.add_argument("--no_cuda", action="store_true", help="Avoid using CUDA when available")