Add file IO

app.py

@@ -13,21 +13,21 @@ def modernisa(lines=None, file_obj=None):
         if is_file_valid:
             with open(file_obj.name) as file:
                 lines = file.read()
-    generated_text = ""
+    generated_text = []
     if lines:
         lines = [line for line in lines.strip().split('\n') if line.strip()]
         text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
         outputs = []
         for line in lines:
             outputs += text2text_generator([line], max_length=150)
-        generated_text = "<br/>".join(output["generated_text"] for output in outputs)
+        generated_text = [output["generated_text"] for output in outputs]
         if is_file_valid and 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
+    with open(output_file, "w") as output:
+        output.write("\n".join(generated_text))
+    return "<br/>".join(generated_text), output
 
 
 samples = [
@@ -55,4 +55,4 @@ gr.Interface(
         gr.outputs.HTML(label="Modern Spanish"),
         gr.outputs.File(label="Download file"),
     ]
-).launch(inline=False)
+).launch(inline=False)