Update app.py

app.py

@@ -1,5 +1,5 @@
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoConfig
-#from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoConfig
+
 import gradio as gr
 from torch.nn import functional as F
 import seaborn
@@ -22,13 +22,13 @@ 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)
+config = AutoConfig.from_pretrained(MODEL_NAME)
 
 MODEL_BUF = {
     "name": MODEL_NAME,
     "tokenizer": tokenizer,
-    "model": model
-    #"config": config
+    "model": model,
+    "config": config
 }
 
 font_dir = ['./']
@@ -44,7 +44,7 @@ def change_model_name(name):
     MODEL_BUF["name"] = name
     MODEL_BUF["tokenizer"] = AutoTokenizer.from_pretrained(name)
     MODEL_BUF["model"] = AutoModelForSeq2SeqLM.from_pretrained(name)
-    #MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
+    MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
 
 
 def generate(text, model, tokenizer):
@@ -75,9 +75,9 @@ app = gr.Interface(
 
     examples = [[MODEL_BUF["name"], text]],
 
-    title="FReE",
+    title="FTG",
 
-    description="Financial relations classifier"
+    description="Financial Text Generation"
 
     )