Update app.py

app.py

@@ -15,7 +15,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)
+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.
@@ -35,7 +35,7 @@ generation_config = GenerationConfig(
     top_p=0.995,      # default 0.75
     top_k=100,        # default 80
     repetition_penalty=1.5,
-    max_new_tokens=2,
+    max_new_tokens=max_tokens,
 )
 
 if instruction: