backup

app.py

@@ -16,17 +16,19 @@ from datasets import load_dataset
 
 api = HfApi(token=os.environ["HF_TOKEN"])
 
-VALID_DATASET = load_dataset("taesiri/IERv2-Subset", split="train")
+VALID_DATASET = load_dataset("taesiri/IERv2-Subset-Validation-150", split="train")
 
 VALID_DATASET_POST_IDS = (
-    load_dataset("taesiri/IERv2-Subset", split="train", columns=["post_id"])
+    load_dataset(
+        "taesiri/IERv2-Subset-Validation-150", split="train", columns=["post_id"]
+    )
     .to_pandas()["post_id"]
     .tolist()
 )
 
 POST_ID_TO_ID_MAP = {post_id: idx for idx, post_id in enumerate(VALID_DATASET_POST_IDS)}
 
-DATASET_REPO = "taesiri/AIImageEditingResults_Intemediate"
+DATASET_REPO = "taesiri/AIImageEditingResults_Intemediate2"
 FINAL_DATASET_REPO = "taesiri/AIImageEditingResults"