import torch
from stimulus_synthesis import NevoPipeline

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

target_vector = torch.zeros(20484)
target_vector[:10] = 1.0
out = pipe(target={"type": "vector", "vector": target_vector.tolist()}, seed_prompts=["natural action video"])
print(out.best_prompt, out.best_score)