Update README.md

README.md

@@ -91,7 +91,9 @@ The model was trained with the parameters:
 
 `torch.utils.data.dataloader.DataLoader` of length 14756 with parameters:
 ```
-{'batch_size': 4, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
+{"multi_target_strategy": "multi-output",
+    "candidate_labels": ['atmosphere',  'beer list',  'bottomless brunch',  'breakfast',  'brunch',  'business',  'byob',  'casual',  'date night',  'delivery',  'fine dining',  'gluten free',  'good value',  'great cocktails',  'great service',  'happy hour',  'healthy',  'instagrammable',  'kid friendly',  'large groups',  'late night',  'live music',  'lunch',  'night out',  'outdoor seating',  'quick',  'romantic',  'rooftop',  'small plates',  'special event',  'sports/tvs',  'takeout',  'tasting menu',  'vegetarian',  'walk-in',  'wine list'],
+   'batch_size': 4, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
 ```
 
 **Loss**: