Update README.md

README.md

@@ -75,10 +75,9 @@ This model was created by using the [llm-compressor](https://github.com/vllm-pro
 
 ```python
 from transformers import AutoTokenizer
-from datasets import Dataset
+from datasets import load_dataset
 from llmcompressor.transformers import SparseAutoModelForCausalLM, oneshot
 from llmcompressor.modifiers.quantization import GPTQModifier
-import random
 
 model_id = "google/gemma-2-2b-it"