Update app.py

app.py

@@ -4,9 +4,9 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 # xl size run out of memory on 16GB vm
 # All the models have input length of 512 tokens and outputs of 512 tokens
-# small 250M param
-# base
-# large
+# small 80M param
+# base 250M
+# large 780M
 # xl
 # xxl
 model_name = "large"
@@ -56,7 +56,7 @@ def text2text(input_text):
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-        # Flan T5 Large Demo
+        # Flan T5 Large Demo (Chat Mode)
         780M parameter Large language model fine tuned on diverse tasks.
         Prompt the model in the Input box. Models output is appended to input. To get additional generation hit submit again.
         """)