Update README.md

README.md

@@ -88,10 +88,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 = "meta-llama/Meta-Llama-3.1-70B-Instruct"