Update README.md

README.md

@@ -75,8 +75,21 @@ pipe = pipeline(
             model=self.model,
             tokenizer=self.tokenizer,
             pipeline_class=AtheneRewardPipeline,
+            device_map="auto",
         )
 
+messages = [
+    {
+        "role": 'user',
+        "content": "What is an Athene Noctura? Explain one sentence."
+    },
+    {
+        "role": "assistant",
+        "content": "The Athene noctua, also known as the little owl, is a small, nocturnal owl species native to Europe, Asia, and North Africa, characterized by its distinctive facial disk and piercing yellow eyes."
+    }
+]
+
+print(pipe([messages])) # Print the reward!
 
 
 ```