Update app.py

app.py

@@ -14,6 +14,7 @@ with col1:
 
 with col2:
     temperature = st.number_input('Temperature: ', min_value=0.0, max_value=1.0, value=0.5, format='%f')
+    max_tokens = st.number_input('Max output length: ', min_value=1, max_value=64, format='%i')
 
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
@@ -24,7 +25,7 @@ model = T5ForConditionalGeneration.from_pretrained(
 )
 
 instruction = st.text_area('Write an instruction:')
-max_tokens = st.number_input('Max output length: ', min_value=1, max_value=64, format='%i')
+
 
 prompts = [
     f"""Below is an instruction that describes a task. Write a response that appropriately completes the request.