Update README.md

README.md

@@ -63,6 +63,15 @@ You can randomly sample one term from each of the nine attribute categories defi
 
 ```python
 # Sampled attributes from UniEM-3M
+from collections import defaultdict
+from datasets import load_dataset
+
+attribute_values = defaultdict(set)
+dataset = load_dataset('NNNan/UniEM-3M')
+
+for attr_dict in dataset['synthesized_data_structured_descriptions']['attribute_description']:
+  for key, value in attr_dict.items():
+    attribute_values[key].add(value)
 
 
 # Construct the prompt