versae commited on
Commit
68e305f
·
1 Parent(s): a941a0d

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import transformers
3
+ from transformers import AutoTokenizer, AutoModel, pipeline, T5ForConditionalGeneration
4
+
5
+ model_name = "versae/modernisa-pre"
6
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
7
+ model = T5ForConditionalGeneration.from_pretrained(model_name)
8
+
9
+ def modernisa(inputs):
10
+ lines = inputs.split('\n')
11
+ text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
12
+ outputs = text2text_generator(lines, max_length=150)
13
+ return "\n".join(output["generated_text"] for output in outputs)
14
+
15
+ gr.Interface(fn=modernisa, inputs=gr.inputs.Textbox(lines=12, label="Enter Old Spanish"), outputs="text").launch(inline=False)