passing caption through json

api/model/src/datasets/dresscode.py

@@ -25,6 +25,7 @@ class DressCodeDataset(data.Dataset):
                  dataroot_path: str,
                  phase: str,
                  tokenizer,
+                 json_from_req,
                  radius=5,
                  caption_folder='fine_captions.json',
                  coarse_caption_folder='coarse_captions.json',
@@ -48,6 +49,7 @@ class DressCodeDataset(data.Dataset):
         self.height = size[0]
         self.width = size[1]
         self.radius = radius
+        self.json_from_req = json_from_req
         self.tokenizer = tokenizer
         self.transform = transforms.Compose([
             transforms.ToTensor(),
@@ -71,12 +73,19 @@ class DressCodeDataset(data.Dataset):
         assert all(x in possible_outputs for x in outputlist)
 
         # Load Captions
+        '''
         with open(self.dataroot / self.caption_folder) as f:
             self.captions_dict = json.load(f)
         self.captions_dict = {k: v for k, v in self.captions_dict.items() if len(v) >= 3}
 
         with open(self.dataroot / coarse_caption_folder) as f:
             self.captions_dict.update(json.load(f))
+        '''
+
+        # Load Captions
+        model_data  = self.json_from_req.get('MODEL', {})  # Safely get the 'MODEL' key, default to an empty dictionary if it doesn't exist
+        # 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}
 
         for c in category:
             assert c in ['dresses', 'upper_body', 'lower_body']