Update app.py

app.py

@@ -61,7 +61,7 @@ def generate_text(model, prompt, max_length=150, temperature=1.0):
         num_return_sequences=1,
         pad_token_id=tokenizer.eos_token_id
     )
-    generated_text = tok.decode(output[0], skip_special_tokens=True)
+    generated_text = textwrap.fill(tok.decode(out_), 60)
     return generated_text