Update README.md

README.md

@@ -44,4 +44,15 @@ predictions = torch.argmax(torch.softmax(logits, dim=1).detach().cpu(), axis=1)
 for input, prediction in zip(input, predictions):
   print(f"task: {task_name}, input: {input} \n prediction=> {prediction}")
   print()
+  
+
+## Statement vs Question Classifier
+input = ["where is berlin?", "is this a keyword query?", "Berlin is in Germany."]
+task_name="spaadia_squad_pairs"
+sequence = tokenizer(input, padding=True, return_tensors="pt")['input_ids']
+logits = model(sequence, task_name=task_name)[0]
+predictions = torch.argmax(torch.softmax(logits, dim=1).detach().cpu(), axis=1)
+for input, prediction in zip(input, predictions):
+  print(f"task: {task_name}, input: {input} \n prediction=> {prediction}")
+  print()
 ```