ManthanKulakarni
commited on
Commit
•
0747d6b
1
Parent(s):
06585ab
Update README.md
Browse files
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=
|
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 |
|