Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def change_model_name(name):
|
|
47 |
MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
|
48 |
|
49 |
|
50 |
-
def generate(
|
51 |
|
52 |
if model_name != MODEL_NAME:
|
53 |
change_model_name(model_name)
|
@@ -71,7 +71,8 @@ if __name__ == '__main__':
|
|
71 |
|
72 |
model_name_list = [
|
73 |
|
74 |
-
'yseop/FNP_T5_D2T_complete'
|
|
|
75 |
|
76 |
]
|
77 |
|
|
|
47 |
MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
|
48 |
|
49 |
|
50 |
+
def generate(modelname, text):
|
51 |
|
52 |
if model_name != MODEL_NAME:
|
53 |
change_model_name(model_name)
|
|
|
71 |
|
72 |
model_name_list = [
|
73 |
|
74 |
+
'yseop/FNP_T5_D2T_complete',
|
75 |
+
'yseop/FNP_T5_D2T_simple'
|
76 |
|
77 |
]
|
78 |
|