Update README.md

README.md

@@ -6,4 +6,24 @@ This is a glorious and graceful gift to the open source community from PyThess m
 
 Built on top of Llama-3.2-1B
 
-Fine tuned with a dataset with sarcastic short "answers" to questions.
+Fine tuned with a dataset with sarcastic short "answers" to questions.
+To test:
+```python
+import torch
+from transformers import pipeline
+
+pipe = pipeline(
+    "text-generation",
+    model="AlexandrosChariton/SarcasLLM-1B",
+    torch_dtype=torch.float32,
+    device_map="auto",
+)
+messages = [
+    {"role": "user", "content": "Why do I even bother with Python? Is it any good?"},
+]
+outputs = pipe(
+    messages,
+    max_new_tokens=128
+)
+print(outputs[0]["generated_text"][-1])
+```