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 |
|