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 |
|