| 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) | |