|
|
|
|
|
import gradio as gr |
|
|
|
models_id = [ |
|
"PORTULAN/albertina-ptpt", |
|
"PORTULAN/albertina-ptbr", |
|
|
|
] |
|
models = list(map(lambda x: gr.load(x, src='huggingface'),models_id)) |
|
examples = [ |
|
[models_id[0],'A culinária portuguesa é rica em sabores e [MASK], tornando-se um dos maiores tesouros do país.'], |
|
[models_id[0],'O clube de futebol português com mais títulos é o [MASK].'], |
|
[models_id[1],'A culinária brasileira é rica em sabores e [MASK], tornando-se um dos maiores patrimônios do país.'], |
|
[models_id[1],'Nascido em Pernambuco, no dia 27 de Outubro 1945, sindicalista e político brasileiro, mais conhecido como [MASK].'], |
|
] |
|
if len(models_id)>2 : |
|
examples.append([models_id[2],'My name is Julien and I like to']) |
|
examples.append([models_id[2],'My name is Clara and I am']) |
|
|
|
def prediction(model_choice, input): |
|
model = models[model_choice] |
|
output = model(input) |
|
return output |
|
|
|
label = gr.Label(num_top_classes=1) |
|
|
|
gr.Interface(fn=prediction, inputs=[ gr.Dropdown(models_id, type="index",value=models_id[0]),gr.Textbox(),], outputs=label, examples=examples).launch() |
|
|
