Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
 tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
 model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
 
-title = "Flan T5 Large Demo"
+title = ""
 
 def get_examples ():
     return [
@@ -23,15 +23,19 @@ on Friday is higher by 40 * 40/100 = <<40*40/100=16>>16 books. How many books do
 def text2text(input_text):
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids
 
-    outputs = model.generate(input_ids, max_length=100)
+    outputs = model.generate(input_ids, max_length=200)
     return tokenizer.decode(outputs[0])
 
 
 with gr.Blocks() as demo:
-
+    gr.Markdown(
+    """
+    # Flan T5 Large Demo
+    Prompt the model in the Input box.
+    """)
     txt_in = gr.Textbox(label="Input", lines=3)
     correct_label = gr.Label(label="Correct")
-    txt_out = gr.Textbox(value="", label="Output", lines=3)
+    txt_out = gr.Textbox(value="", label="Output", lines=4)
     
 
     btn = gr.Button(value="Submit")
@@ -43,7 +47,6 @@ with gr.Blocks() as demo:
         inputs=[txt_in,correct_label]
     )
     
-demo.
 
 if __name__ == "__main__":
     demo.launch()