Update README.md

README.md

@@ -10,7 +10,9 @@ widget:
 This is the model used for knowledge clustering where we feed title-body pair and the classifier predicts if the pair is valid or not.
 For further information, please refer to https://github.com/yctam/dstc10_track2_task2 for the Github repository.
 
-## Usage
+
+---
+```python
 def single_test(tokenizer, title_body_pair):
     result = tokenizer([title_body_pair], return_tensors="pt")
     result.to(device)
@@ -24,4 +26,4 @@ if __name__ == '__main__':
     body = "Yes, 24 Hours Front Desk Avaliable."
     pair = (sentence, body)
     print(single_test(pair)) # Expect: True       
----
+```