from stimulus_synthesis import NevoPipeline

pipe = NevoPipeline.from_pretrained(
    "epfl-neuroai/NEvo",
    synthesis_config={
        "text_to_image_model_id": "stabilityai/sdxl-turbo",
        "image_to_video_model_id": "Lightricks/LTX-Video-0.9.8-13B-distilled",
        "encoder_model_id": "epfl-neuroai/vjepa2-encoder-basic",
    },
)

out = pipe(target={"type": "indices", "indices": [0, 1, 2]}, seed_prompts=["a moving person"])
print(out.best_prompt, out.best_score)