Update app.py

app.py

@@ -22,8 +22,6 @@ import matplotlib.font_manager as fm
 MODEL_NAME = 'https://huggingface.co/yseop/FNP_T5_D2T_complete'
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
-
-
 config = AutoConfig.from_pretrained(MODEL_NAME)
 
 MODEL_BUF = {
@@ -51,7 +49,7 @@ def change_model_name(name):
 
 def generate(text, model, tokenizer):
    model.eval()
-   input_ids = tokenizer.encode("webNLG:{}".format(text), return_tensors="pt")
+   input_ids = tokenizer.encode("AFA:{}".format(text), return_tensors="pt")
    outputs = model.generate(input_ids, max_length=200, num_beams=2, repetition_penalty=2.5, top_k=50, top_p=0.98, length_penalty=1.0, early_stopping=True)
    return tokenizer.decode(outputs[0])
 
@@ -60,7 +58,7 @@ def generate(text, model, tokenizer):
 
 if __name__ == '__main__':
 
-    text = 'Group profit | valIs | € 115.7 million && € 115.7 million | dTime | in 2019'
+    text = ['Group profit | valIs | € 115.7 million && € 115.7 million | dTime | in 2019']
 
     model_name_list = [