app

app.py

@@ -25,7 +25,7 @@ def parse_args(args):
     parser.add_argument('--mode', type=str, default="train")
     parser.add_argument('--scale', type=bool, default=True)
     parser.add_argument('--smote', type=bool, default=True)
-    parser.add_argument('--model_name_or_path', type=str, default="./ensemble_1", choices=[])
+    parser.add_argument('--model_name_or_path', type=str, default="./FINAL", choices=[])
     parser.add_argument('--top_p', type=float, default=None)
     parser.add_argument('--temperature', type=float, default=1.0)
     parser.add_argument('--device', type=str, default="cuda", choices=["cuda", "cpu"])