Create README.md

README.md

@@ -0,0 +1,26 @@
+
+
+## Model in Action 🚀
+
+```python
+from transformers import AutoModelWithLMHead, AutoTokenizer
+
+tokenizer = AutoTokenizer.from_pretrained("ManthanKulakarni/Text2JQLBuilder_v2")
+model = AutoModelWithLMHead.from_pretrained("ManthanKulakarni/Text2JQLBuilder_v2")
+
+def gen_sentence(words, max_length=32):
+  input_text = words
+  features = tokenizer([input_text], return_tensors='pt')
+
+  output = model.generate(input_ids=features['input_ids'], 
+               attention_mask=features['attention_mask'],
+               max_length=max_length)
+
+  return tokenizer.decode(output[0], skip_special_tokens=True)
+
+words = "JQL: all story under project ACM"
+
+gen_sentence(words)
+
+# output: 'JQL: project = ACM'
+```