Update app.py

app.py

@@ -7,19 +7,24 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
 
 def modernisa(lines=None, file_obj=None):
+    is_file_valid = False
     if file_obj:
-        with open(file_obj.name) as file:
-            lines = file.read()
-    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)
-    if file_obj:
-        output_file = f"{file_obj.name.rsplit('.', 1)[0]}_modernized.txt"
-    else:
-        output_file = "modernized.txt"
+        is_file_valid = file_obj.name.endswith(".txt")
+        if is_file_valid:
+            with open(file_obj.name) as file:
+                lines = file.read()
+    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)
+        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
@@ -43,11 +48,11 @@ porque el pensamiento creo"""
 gr.Interface(
     fn=modernisa,
     inputs=[
-        gr.inputs.Textbox(lines=12, label="Enter Old Spanish", default=random.choice(samples)),
+        gr.inputs.Textbox(lines=12, label="Enter Spanish Golden Age text", 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"),
-        gr.outputs.File(label="Download file")
+        gr.outputs.File(label="Download file"),
     ]
 ).launch(inline=False)