Update app.py

app.py

@@ -20,7 +20,7 @@ def generate_text(input_text):
     # generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
 
     output = model.generate(input_text)
-    return generated_text
+    return output
 
 text_generation_interface = gr.Interface(
     fn=generate_text,