Spaces:
Sleeping
Sleeping
change default of --model_dir
Browse files
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="./
|
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")
|