Update README.md

README.md

@@ -16,6 +16,7 @@ The easiest way to use our models is through the `rerankers` package. After inst
 
 ```python
 from rerankers import Reranker
+import torch
 
 query = "O futebol é uma paixão nacional"
 docs = [
@@ -24,54 +25,35 @@ docs = [
 ]
 
 ranker = Reranker(
-    "unicamp-dl/monoptt5-small",
+    "unicamp-dl/monoptt5-base",
     inputs_template="Pergunta: {query} Documento: {text} Relevante:",
+    dtype=torch.float32  # or bfloat16 if supported by your GPU
 )
-# Relevant logging:
-# Loading T5Ranker model unicamp-dl/monoptt5-small
+
+results = ranker.rank(query, docs)
+
+print("Classification results:")
+for result in results:
+    print(result)
+
+# Loading T5Ranker model unicamp-dl/monoptt5-base
 # No device set
-# Using device cpu
-# No dtype set
-# Device set to `cpu`, setting dtype to `float32`
+# Using device cuda
 # Using dtype torch.float32
-# Loading model unicamp-dl/monoptt5-small, this might take a while...
-# Using device cpu.
+# Loading model unicamp-dl/monoptt5-base, this might take a while...
+# Using device cuda.
 # Using dtype torch.float32.
 # T5 true token set to ▁Sim
 # T5 false token set to ▁Não
 # Returning normalised scores...
 # Inputs template set to Pergunta: {query} Documento: {text} Relevante:
 
-results = ranker.rerank(query, docs)
-# Results should be something like (can vary depending on the model, the example below uses the "unicamp-dl/monoptt5-small" model)
-RankedResults(
-    results=[
-        Result(
-            document=Document(
-                text="O futebol é uma parte essencial da cultura brasileira e une as pessoas.",
-                doc_id=1,
-                metadata={},
-            ),
-            score=0.91943359375,
-            rank=1,
-        ),
-        Result(
-            document=Document(
-                text="O futebol é superestimado e não deveria receber tanta atenção.",
-                doc_id=0,
-                metadata={},
-            ),
-            score=0.0267486572265625,
-            rank=2,
-        ),
-    ],
-    query="O futebol é uma paixão nacional",
-    has_scores=True,
-)
-
+# Classification results:
+# document=Document(text='O futebol é uma parte essencial da cultura brasileira e une as pessoas.', doc_id=1, metadata={}) score=0.8186910152435303 rank=1
+# document=Document(text='O futebol é superestimado e não deveria receber tanta atenção.', doc_id=0, metadata={}) score=0.008028557524085045 rank=2
 ```
 
-For additional configurations and more advanced usage, consult the rerankers documentation.
+For additional configurations and more advanced usage, consult the `rerankers` [GitHub repository](https://github.com/AnswerDotAI/rerankers).
 
 # Citation
 If you use our models, please cite: