Update app.py

app.py

@@ -6,8 +6,11 @@ model_name = "versae/byt5-base-finetuned-modernisa"  # "versae/modernisa-pre"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
 
-def modernisa(inputs):
-    lines = [line for line in inputs.strip().split('\n') if line.strip()]
+def modernisa(lines=None, file_obj=None):
+    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:
@@ -32,6 +35,9 @@ porque el pensamiento creo"""
 
 gr.Interface(
     fn=modernisa,
-    inputs=gr.inputs.Textbox(lines=12, label="Enter Old Spanish", default=random.choice(samples)),
+    inputs=[
+        gr.inputs.Textbox(lines=12, label="Enter Old Spanish", default=random.choice(samples), optional=True),
+        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"),
 ).launch(inline=False)