| import argparse | |
| from model_loader import load_model | |
| from generator import generate_code | |
| from utils import pretty_print_code | |
| if __name__ == "__main__": | |
| parser = argparse.ArgumentParser(description="Run Sentinel for code generation") | |
| parser.add_argument("prompt", type=str, help="The coding prompt to generate code for") | |
| parser.add_argument("--max_new_tokens", type=int, default=200, help="Maximum tokens to generate") | |
| args = parser.parse_args() | |
| generator = load_model("your-username/sentinel") | |
| code = generate_code(args.prompt, generator, max_new_tokens=args.max_new_tokens) | |
| pretty_print_code(code) | |