Update app.py

app.py

@@ -1,9 +1,10 @@
 import gradio as gr
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-base")
-model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
+tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
+model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
 
+title = "Flan T5 Large Demo"
 
 def get_examples ():
     return [
@@ -22,15 +23,15 @@ 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=40)
+    outputs = model.generate(input_ids, max_length=100)
     return tokenizer.decode(outputs[0])
 
 
 with gr.Blocks() as demo:
 
-    txt_in = gr.Textbox(label="Input", lines=2)
+    txt_in = gr.Textbox(label="Input", lines=3)
     correct_label = gr.Label(label="Correct")
-    txt_out = gr.Textbox(value="", label="Output")
+    txt_out = gr.Textbox(value="", label="Output", lines=3)
     
 
     btn = gr.Button(value="Submit")
@@ -41,7 +42,8 @@ with gr.Blocks() as demo:
         examples=get_examples(),
         inputs=[txt_in,correct_label]
     )
-
+    
+demo.
 
 if __name__ == "__main__":
     demo.launch()