DebeshSahoo commited on
Commit
b74c191
·
1 Parent(s): 052890e
Files changed (1) hide show
  1. README.md +5 -2
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("t5-small")
14
 
15
  # Load the model
16
- model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
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