Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
1e660fe
1
Parent(s): a208e00
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
|
@@ -5,4 +5,5 @@ import pandas as pd
|
|
| 5 |
from datasets import load_metric
|
| 6 |
from transformers import (
|
| 7 |
AdamW,
|
|
|
|
| 8 |
T5TokenizerFast as T5Tokenizer,
|
|
|
|
| 5 |
from datasets import load_metric
|
| 6 |
from transformers import (
|
| 7 |
AdamW,
|
| 8 |
+
T5ForConditionalGeneration,
|
| 9 |
T5TokenizerFast as T5Tokenizer,
|