Fixed json req

api/model/src/datasets/vitonhd.py

@@ -78,7 +78,7 @@ class VitonHDDataset(data.Dataset):
         #self.captions_dict = {k: v for k, v in self.captions_dict.items() if len(v) >= 3}
 
         # 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
+        model_data  = self.json_from_req.get('vitonhd', {})  # Safely get the 'vitonhd' 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}