ManthanKulakarni commited on
Commit
0747d6b
1 Parent(s): 06585ab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ from transformers import AutoModelWithLMHead, AutoTokenizer
24
  tokenizer = AutoTokenizer.from_pretrained("ManthanKulakarni/Text2JQLBuilder_v2")
25
  model = AutoModelWithLMHead.from_pretrained("ManthanKulakarni/Text2JQLBuilder_v2")
26
 
27
- def gen_sentence(words, max_length=32):
28
  input_text = words
29
  features = tokenizer([input_text], return_tensors='pt')
30
 
 
24
  tokenizer = AutoTokenizer.from_pretrained("ManthanKulakarni/Text2JQLBuilder_v2")
25
  model = AutoModelWithLMHead.from_pretrained("ManthanKulakarni/Text2JQLBuilder_v2")
26
 
27
+ def gen_sentence(words, max_length=64):
28
  input_text = words
29
  features = tokenizer([input_text], return_tensors='pt')
30