Update app.py
Browse files
app.py
CHANGED
@@ -20,11 +20,10 @@ import matplotlib.font_manager as fm
|
|
20 |
# global var
|
21 |
|
22 |
MODEL_NAME = 'https://huggingface.co/yseop/FNP_T5_D2T_complete'
|
23 |
-
|
24 |
-
|
|
|
25 |
|
26 |
-
tokenizer = T5Tokenizer.from_pretrained(MODEL_NAME)
|
27 |
-
model = T5ForConditionalGeneration.from_pretrained(MODEL_NAME)
|
28 |
config = AutoConfig.from_pretrained(MODEL_NAME)
|
29 |
|
30 |
MODEL_BUF = {
|
|
|
20 |
# global var
|
21 |
|
22 |
MODEL_NAME = 'https://huggingface.co/yseop/FNP_T5_D2T_complete'
|
23 |
+
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
24 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
|
25 |
+
|
26 |
|
|
|
|
|
27 |
config = AutoConfig.from_pretrained(MODEL_NAME)
|
28 |
|
29 |
MODEL_BUF = {
|