rewriting file text

api/model/src/datasets/vitonhd.py

@@ -82,6 +82,13 @@ class VitonHDDataset(data.Dataset):
         # Filter captions based on the length requirement (3 or more items)
         self.captions_dict = {k: v for k, v in model_data.items() if len(v) >= 3}
 
+        pairs_file_path = os.path.join(self.dataroot, f"{self.phase}_pairs.txt")
+
+
+        with open(pairs_file_path, 'w') as pairs_file:
+            for key in self.captions_dict.keys():
+                pairs_file.write(f"{key}_00.jpg {key}_00.jpg\n")
+
         dataroot = self.dataroot
         if phase == 'train':
             filename = os.path.join(dataroot, f"{phase}_pairs.txt")