Update README.md

README.md

@@ -33,6 +33,33 @@ Raises:
 
 For more information on my dataset, please see the included referenced dataset.
 
+You can test the model using this:
+
+```
+from transformers import T5ForConditionalGeneration, AutoTokenizer
+
+checkpoint = "Mir-2002/codet5p-google-style-docstrings"
+device = "cuda" # or CPU
+
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+model = T5ForConditionalGeneration.from_pretrained(checkpoint).to(device)
+
+input = """
+def calculate_sum(a, b):
+    return a + b
+"""
+
+inputs = tokenizer.encode(input, return_tensors="pt").to(device)
+outputs = model.generate(inputs, max_length=128)
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+# Calculate the sum of two numbers.
+
+# Args:
+# a (int): The first number.
+# b (int): The second number.
+
+```
+
 # Hyperparameters
 
 MAX_SOURCE_LENGTH = 256 <br>