3_start_prodigy_UI.py

import subprocess
import argparse


# Initialize argparse to parse command-line arguments
parser = argparse.ArgumentParser(description="Run a Prodigy custom-text-classification task.")
parser.add_argument("--dataset", required=True, help="The name of the dataset.")
parser.add_argument("--input_file", required=True, help="The path to the input file on the local prodigy machine.")

# Parse the arguments
args = parser.parse_args()



# Your command broken down into its individual components, with the timestamp
command = [
    "prodigy",
    "custom-text-classification",  # task name/recipe/layout
    args.dataset,  # dataset name from command line args
    args.input_file,  # input file path from command line args
    "-F",
    "scripts/custom_task_template.py"  # layout source
]

# Run the command
subprocess.run(command)