Update README.md

README.md

@@ -49,6 +49,9 @@ $$\text{S}_{predicted} \times (\text{max} - \text{min}) + \text{min}$$
 Use the code below to get started with the model.
 
 ```python
+import torch
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
 model_name = "thethinkmachine/Maxwell-Task-Complexity-Scorer-v0.2"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -77,8 +80,8 @@ query = "Is learning equivalent to decreasing local entropy?"
 max_score = 100
 min_score = 0
 
-print("DEITA Evol-Complexity Score:", get_deita_complexity_score(query)) # 2
-print("Scaled Complexity Score:", get_scaled_complexity_score(query)) # 28.39...
+print("DEITA Evol-Complexity Score:", get_deita_complexity_score(query))
+print("Scaled Complexity Score:", get_scaled_complexity_score(query))
 ```
 
 ## Training Details