fkalpana commited on
Commit
fc83495
Β·
verified Β·
1 Parent(s): 04adb18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
3
  from datasets import load_dataset
4
 
5
  # Load the tokenizer and model
6
- tokenizer = T5Tokenizer.from_pretrained('t5-small')
7
- model = T5ForConditionalGeneration.from_pretrained('t5-small', legacy=False)
8
 
9
  dataset = load_dataset("wikisql", split='test')
10
  for i in range(3): # Let's take the first 3 examples
 
3
  from datasets import load_dataset
4
 
5
  # Load the tokenizer and model
6
+ tokenizer = T5Tokenizer.from_pretrained('t5-small', legacy=False)
7
+ model = T5ForConditionalGeneration.from_pretrained('t5-small')
8
 
9
  dataset = load_dataset("wikisql", split='test')
10
  for i in range(3): # Let's take the first 3 examples