File size: 1,222 Bytes
833e26d cf39375 833e26d f11101d 833e26d f11101d 833e26d cf39375 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# https://stackoverflow.com/questions/4506758/flatten-old-history-in-git
import gradio as gr
models_id = [
"PORTULAN/albertina-ptpt",
"PORTULAN/albertina-ptbr",
# "EleutherAI/gpt-j-6B"
]
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()
|