ManthanKulakarni commited on
Commit
0252aad
1 Parent(s): b0a887c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -10,6 +10,7 @@ tags:
10
  - JQL
11
  - JQL Builder
12
  - text2jql
 
13
  ---
14
 
15
 
@@ -21,7 +22,7 @@ from transformers import AutoModelWithLMHead, AutoTokenizer
21
  tokenizer = AutoTokenizer.from_pretrained("ManthanKulakarni/Text2JQLBuilder")
22
  model = AutoModelWithLMHead.from_pretrained("ManthanKulakarni/Text2JQLBuilder")
23
 
24
- def gen_sentence(words, max_length=32):
25
  input_text = words
26
  features = tokenizer([input_text], return_tensors='pt')
27
 
@@ -36,4 +37,4 @@ words = "JQL: all story under project ACM"
36
  gen_sentence(words)
37
 
38
  # output: 'JQL: project = ACM'
39
- ```
 
10
  - JQL
11
  - JQL Builder
12
  - text2jql
13
+ - Flan-T5
14
  ---
15
 
16
 
 
22
  tokenizer = AutoTokenizer.from_pretrained("ManthanKulakarni/Text2JQLBuilder")
23
  model = AutoModelWithLMHead.from_pretrained("ManthanKulakarni/Text2JQLBuilder")
24
 
25
+ def gen_sentence(words, max_length=64):
26
  input_text = words
27
  features = tokenizer([input_text], return_tensors='pt')
28
 
 
37
  gen_sentence(words)
38
 
39
  # output: 'JQL: project = ACM'
40
+ ```