Update app.py

app.py

@@ -13,10 +13,10 @@ story = pipeline(
 )
 
 
-def generate(tags_text, temp=1.0, n_beams=3, max_l=128):
+def generate(tags_text, temp=1.0, n_beams=3):
     tags = tags_text.split(", ")
     prefix = f"~^{'^'.join(tags)}~@"
-    g_text = story(prefix, temperature=1.0, max_length=128, repetition_penalty=7.0, num_beams=3)[0]['generated_text']
+    g_text = story(prefix, temperature=1.0, repetition_penalty=7.0, num_beams=3)[0]['generated_text']
     return g_text[g_text.find("@") + 1:]