Text2JQLBuilder / README.md
ManthanKulakarni's picture
Update README.md
0252aad
|
raw
history blame
860 Bytes
metadata
license: bsd
datasets:
  - ManthanKulakarni/Text2JQL
language:
  - en
pipeline_tag: text2text-generation
tags:
  - Jira
  - JQL
  - JQL Builder
  - text2jql
  - Flan-T5

Model in Action πŸš€

from transformers import AutoModelWithLMHead, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("ManthanKulakarni/Text2JQLBuilder")
model = AutoModelWithLMHead.from_pretrained("ManthanKulakarni/Text2JQLBuilder")

def gen_sentence(words, max_length=64):
  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'