Update app.py
Browse files
app.py
CHANGED
@@ -22,13 +22,13 @@ import matplotlib.font_manager as fm
|
|
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 |
-
config = AutoConfig.from_pretrained(MODEL_NAME)
|
26 |
|
27 |
MODEL_BUF = {
|
28 |
"name": MODEL_NAME,
|
29 |
"tokenizer": tokenizer,
|
30 |
-
"model": model
|
31 |
-
"config": config
|
32 |
}
|
33 |
|
34 |
font_dir = ['./']
|
@@ -44,7 +44,7 @@ def change_model_name(name):
|
|
44 |
MODEL_BUF["name"] = name
|
45 |
MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
|
46 |
MODEL_BUF["model"] = AutoModelForSeq2SeqLM.from_pretrained(name)
|
47 |
-
MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
|
48 |
|
49 |
|
50 |
def generate(text, model, tokenizer):
|
|
|
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 |
+
#config = AutoConfig.from_pretrained(MODEL_NAME)
|
26 |
|
27 |
MODEL_BUF = {
|
28 |
"name": MODEL_NAME,
|
29 |
"tokenizer": tokenizer,
|
30 |
+
"model": model
|
31 |
+
#"config": config
|
32 |
}
|
33 |
|
34 |
font_dir = ['./']
|
|
|
44 |
MODEL_BUF["name"] = name
|
45 |
MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
|
46 |
MODEL_BUF["model"] = AutoModelForSeq2SeqLM.from_pretrained(name)
|
47 |
+
#MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
|
48 |
|
49 |
|
50 |
def generate(text, model, tokenizer):
|