Update README.md

README.md

@@ -15,28 +15,28 @@ pip install lens-metric
 ```
 
 ```python
-from lens import download_model
-from lens.lens_score import LENS
-
-model_path = download_model("davidheineman/lens")
-lens = LENS(model_path)
-
-score = lens.score(
-    complex = [
-        "They are culturally akin to the coastal peoples of Papua New Guinea."
-    ],
-    simple = [
-        "They are culturally similar to the people of Papua New Guinea."
-    ],
-    references = [
-        [
-            "They are culturally similar to the coastal peoples of Papua New Guinea.",
-            "They are similar to the Papua New Guinea people living on the coast."
-        ]
-    ]
-)
+from lens import download_model, LENS
+
+lens_path = download_model("davidheineman/lens")
+lens = LENS(lens_path, rescale=True)
+
+complex = [
+    "They are culturally akin to the coastal peoples of Papua New Guinea."
+]
+simple = [
+    "They are culturally similar to the people of Papua New Guinea."
+]
+references = [[
+    "They are culturally similar to the coastal peoples of Papua New Guinea.",
+    "They are similar to the Papua New Guinea people living on the coast."
+]]
+
+scores = lens.score(complex, simple, references, batch_size=8, devices=[0])
+print(scores) # [78.6344531130125]
 ```
 
+For an example, please see the [quick demo Google Collab notebook](https://colab.research.google.com/drive/1rIYrbl5xzL5b5sGUQ6zFBfwlkyIDg12O?usp=sharing).
+
 ## Intended uses
 This model is for reference-based text simplification evaluation, for a model requiring no references, please see [**LENS-SALSA**](https://huggingface.co/davidheineman/lens-salsa).