Update README.md

README.md

@@ -8,7 +8,7 @@ language:
 # What does this model do?
 This model converts the natural language input to MongoDB (MQL) query. It is a fine-tuned CodeT5+ 220M. This model is a part of nl2query repository which is present at https://github.com/Chirayu-Tripathi/nl2query
 
-You can use this model via the github repository or via following code.
+You can use this model via the github repository or via following code. More information can be found on the repository.
 
 ```python
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
@@ -17,6 +17,8 @@ model = AutoModelForSeq2SeqLM.from_pretrained("Chirayu/nl2mongo")
 tokenizer = AutoTokenizer.from_pretrained("Chirayu/nl2mongo")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)
+
+textual_query = '''mongo: which cabinet has average age less than 21? | titanic : _id, passengerid, survived, pclass, name, sex, age, sibsp, parch, ticket, fare, cabin, embarked'''
 def generate_query(
         textual_query: str,
         num_beams: int = 10,