Commit
·
b74c191
1
Parent(s):
052890e
update
Browse files
README.md
CHANGED
|
@@ -7,13 +7,16 @@ pipeline_tag: text-classification
|
|
| 7 |
tags:
|
| 8 |
- code
|
| 9 |
---
|
|
|
|
|
|
|
|
|
|
| 10 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 11 |
|
| 12 |
# Load the tokenizer
|
| 13 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 14 |
|
| 15 |
# Load the model
|
| 16 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("
|
| 17 |
|
| 18 |
# Rest of the code for preparing input, generating predictions, and decoding the output...
|
| 19 |
|
|
|
|
| 7 |
tags:
|
| 8 |
- code
|
| 9 |
---
|
| 10 |
+
|
| 11 |
+
#Copy below piece of code to your notebook to use the model
|
| 12 |
+
|
| 13 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 14 |
|
| 15 |
# Load the tokenizer
|
| 16 |
+
tokenizer = AutoTokenizer.from_pretrained("DebeshSahoo/text2sql-finetune")
|
| 17 |
|
| 18 |
# Load the model
|
| 19 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("DebeshSahoo/text2sql-finetune")
|
| 20 |
|
| 21 |
# Rest of the code for preparing input, generating predictions, and decoding the output...
|
| 22 |
|