Update README.md

README.md

@@ -35,12 +35,10 @@ Let us setup the imports, load the model and a utility function to parse the mod
 ```python
 import os, re
 
-from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModel
+from transformers import AutoTokenizer
 import torch
-from torch.nn.functional import softmax
-import jinja2, json
+import json
 from vllm import LLM, SamplingParams
-import math
 
 safe_token = "no"
 risky_token = "yes"