update app.py

app.py

@@ -17,10 +17,10 @@ if torch.cuda.is_available():
 generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device=model.device)
 print('{}:done.'.format(datetime.now().strftime('%Y-%m-%d %H:%M:%S')))
 
-def generate(input_text):
+def generate(input_text, maxlen):
 	output = generator(
 		input_text,
-		max_length=128,
+		max_length=maxlen,
 		do_sample=True,
 		num_return_sequences=1,
 		pad_token_id=tokenizer.pad_token_id,
@@ -35,15 +35,16 @@ with gr.Blocks(title='text generation ja') as chatbox:
 
 	chatbot = gr.Chatbot(label='generated text')
 	msg = gr.Textbox(label='text')
+	maxlen = gr.Slider(minimum=30, maximum=256, value=30, step=1, label='max length')
 	clear = gr.ClearButton([msg, chatbot])
 
-	def respond(message, chat_history):
+	def respond(message, maxlen, chat_history):
 		if message == '':
 			return '', chat_history
-		bot_message = generate(message)
+		bot_message = generate(message, maxlen)
 		chat_history.append((message, bot_message))
 		return '', chat_history
 
-	msg.submit(respond, [msg, chatbot], [msg, chatbot])
+	msg.submit(respond, [msg, maxlen, chatbot], [msg, chatbot])
 
 chatbox.launch()