from stimulus_synthesis import NevoPipeline

pipe = NevoPipeline.from_pretrained("epfl-neuroai/NEvo")

out = pipe(
    target={"type": "indices", "indices": [0, 1, 2]},
    seed_prompts=["a person running through a crowded street"],
    num_candidates=16,
    num_rounds=50,
)
print(out.best_prompt, out.best_score)