Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,8 +15,8 @@ st.set_page_config(
 # Load Model
 # =========================
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-Large")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-Large")
     return tokenizer, model
  
 tokenizer, model = load_model()
@@ -155,7 +155,7 @@ Day 1:
 
                 **inputs,
 
-                max_new_tokens=900,
+                max_new_tokens=600,
 
                 temperature=0.7,