make argparse nicer and fix reference to model

run_model.py

@@ -160,20 +160,28 @@ def run_model_on_data(model_path, tokenizer_name, arguments):
 if __name__ == "__main__":
     # Parse the arguments
     parser = argparse.ArgumentParser()
-    parser.add_argument("--task", type=int, help="The task type (1, 2, or 3)")
+    parser.add_argument(
+        "--task",
+        type=int,
+        choices=(1,2,3),
+        required=True,
+        help="The task type (1, 2, or 3)",
+    )
     parser.add_argument(
         "--input-file",
         type=str,
+        required=True,
         help="The path to the input file with data to be corrected",
     )
     parser.add_argument(
         "--output-file",
         type=str,
+        required=True,
         help="The path to the output file where the corrected data will be saved",
     )
     args = parser.parse_args()
 
-    model_path = "./gpt-sw3-model"
+    model_path = "."
     tokenizer_name = "AI-Sweden-Models/gpt-sw3-6.7b"
 
     set_seed(42)