Update app.py

app.py

@@ -13,10 +13,10 @@ story = pipeline(
 )
 
 
-def generate(tags_text, temp=1.0, n_beams=3):
+def generate(tags_text, temp, n_beams, max_len):
     tags = tags_text.split(", ")
     prefix = f"~^{'^'.join(tags)}~@"
-    g_text = story(prefix, temperature=1.0, repetition_penalty=7.0, num_beams=3)[0]['generated_text']
+    g_text = story(prefix, temperature=temp, repetition_penalty=7.0, num_beams=n_beams, max_length=max_len)[0]['generated_text']
     return g_text[g_text.find("@") + 1:]
 
 
@@ -30,7 +30,8 @@ iface = gr.Interface(generate,
 	inputs = [
     gr.Textbox(label="Keywords (comma separated)"),
 	gr.inputs.Slider(0, 2, default=1.0, step=0.05, label="Temperature"),
-	gr.inputs.Slider(1, 10, default=3, label="Number of beams", step=1)
+	gr.inputs.Slider(1, 10, default=3, label="Number of beams", step=1),
+    gr.Number(label="Max lenght", value=128)
 	],
 	outputs = gr.Textbox(label="Output"),
 	title=title,