Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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'
|
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
|