Update app.py

app.py

@@ -15,7 +15,14 @@ def modernisa(lines=None, file_obj=None):
     outputs = []
     for line in lines:
         outputs += text2text_generator([line], max_length=150)
-    return "<br/>".join(output["generated_text"] for output in outputs)
+    generated_text = "<br/>".join(output["generated_text"] for output in outputs)
+    if file_obj:
+        output_file = f"{file_obj.name.rsplit('.', 1)[0]}_modernized.txt"
+    else:
+        output_file = "modernized.txt"
+    with open(output_file) as output:
+        output.write(generated_text)
+    return generated_text, output
 
 
 samples = [
@@ -36,8 +43,11 @@ porque el pensamiento creo"""
 gr.Interface(
     fn=modernisa,
     inputs=[
-        gr.inputs.Textbox(lines=12, label="Enter Old Spanish", default=random.choice(samples), optional=True),
+        gr.inputs.Textbox(lines=12, label="Enter Old Spanish", default=random.choice(samples)),
         gr.inputs.File(file_count="single", label="Or upload a plain text file (.txt)", type="file", optional=True),
     ],
-    outputs=gr.outputs.HTML(label="Modern Spanish"),
+    outputs=[
+        gr.outputs.HTML(label="Modern Spanish"),
+        gr.outputs.File(label="Download file")
+    ]
 ).launch(inline=False)