diff --git a/example_03_chaos_mode.py b/example_03_chaos_mode.py index 792e71bb..d84b8055 100644 --- a/example_03_chaos_mode.py +++ b/example_03_chaos_mode.py @@ -3,12 +3,20 @@ from wfgy_core import WFGYRunner if __name__ == "__main__": parser = argparse.ArgumentParser() - parser.add_argument("--reflect", action="store_true", help="Enable self-reflection") - parser.add_argument("--ascii", action="store_true", help="Show ASCII banner") - parser.add_argument("--style", type=str, default="meme", help="Choose style: default/scientific/meme") - + parser.add_argument("--reflect", action="store_true", help="Enable self-reflection output") + parser.add_argument("--ascii", action="store_true", help="Enable ASCII visual output") + parser.add_argument("--style", type=str, default="chaos", help="Output style mode") + parser.add_argument("--model", type=str, default="gpt2-medium", help="Model ID (e.g., gpt2, gpt2-xl)") args = parser.parse_args() + config = { + "style": args.style, + "reflect": args.reflect, + "ascii": args.ascii, + "model_id": args.model + } + + runner = WFGYRunner(config=config) prompt = "What happens if Schrödinger’s cat becomes a quantum influencer?" - runner = WFGYRunner() - runner.run(prompt, reflect=args.reflect, style=args.style, show_ascii=args.ascii) + output = runner.run(prompt) + print(output)