Update app.py

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-xl")
-model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl")
+# xl size run out of memory on 16GB vm
+tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
+model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
 
 title = ""
 
@@ -31,6 +32,7 @@ with gr.Blocks() as demo:
     gr.Markdown(
     """
     # Flan T5 Large Demo
+    780M parameter Large language model fine tuned on diverse tasks.
     Prompt the model in the Input box.
     """)
     txt_in = gr.Textbox(label="Input", lines=3)