Downloads · 30 days
21
21% of all-time downloads
ManthanKulakarni/Text2JQLBuilder_v2
Text2JQLBuilder_v2 is a text generation model from ManthanKulakarni. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as bsd.
Downloads · 30 days
21
21% of all-time downloads
All-time downloads
100
Public
Repo size
9.4 GB
Likes
0
Public
Click a slice to open those files.
.bin3.1 GB · 100%
From the Hugging Face model README
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=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'