Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="
|
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="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")
|