Update app.py

app.py

@@ -12,7 +12,7 @@ generator = pipeline(
 )
 model = AutoModelWithLMHead.from_pretrained("gpt2")
 def green(prompt):
-    return generator(prompt)
+    return generator(prompt, max_length=500)["generated_text"]
     
 iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs=gr.outputs.Textbox(type="str", label="generated text"))
 iface.launch()