|
--- |
|
license: bsd |
|
datasets: |
|
- ManthanKulakarni/Text2JQL |
|
language: |
|
- en |
|
pipeline_tag: text2text-generation |
|
tags: |
|
- Jira |
|
- JQL |
|
- JQL Builder |
|
- text2jql |
|
--- |
|
|
|
|
|
## Model in Action 🚀 |
|
|
|
```python |
|
from transformers import AutoModelWithLMHead, AutoTokenizer |
|
|
|
tokenizer = AutoTokenizer.from_pretrained("ManthanKulakarni/Text2JQLBuilder") |
|
model = AutoModelWithLMHead.from_pretrained("ManthanKulakarni/Text2JQLBuilder") |
|
|
|
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' |
|
``` |
|
|
