Spaces:
Running
Running
File size: 792 Bytes
2c50826 34046e2 2c50826 34046e2 2c50826 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
from pathlib import Path
from typing import Iterator, List, Tuple
from datasets import load_dataset
class PartiPrompts:
def __init__(self):
dataset = load_dataset("nateraw/parti-prompts")["train"]
shuffled_dataset = dataset.shuffle(seed=42)
selected_dataset = shuffled_dataset.select(range(800))
self.prompts = [row["Prompt"] for row in selected_dataset]
def __iter__(self) -> Iterator[Tuple[str, Path]]:
for i, prompt in enumerate(self.prompts):
yield prompt, Path(f"{i}.png")
@property
def name(self) -> str:
return "parti"
@property
def size(self) -> int:
return len(self.prompts)
@property
def metrics(self) -> List[str]:
return ["arniqa", "clip", "clip_iqa", "sharpness"]
|