Update app.py

app.py

@@ -4,7 +4,13 @@ from transformers import AutoTokenizer, T5Tokenizer, T5ForConditionalGeneration,
 
 st.title('How does the LLM choose its words?')
 
-model_checkpoint = "google/flan-t5-xl"
+model_checkpoint = st.selectbox(
+    "Model:",
+    ("google/flan-t5-base", "google/flan-t5-xl"),
+    label_visibility=st.session_state.visibility,
+    disabled=st.session_state.disabled,
+)
+temperature = st.number_input('Temperature: ', min_value=0, max_value=1, format='%f')
 
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)