dmariko commited on
Commit
8165996
1 Parent(s): 5963b0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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(model, text):
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