# 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, enable_queue=True, inputs=[ gr.Dropdown(models_id, type="index",value=models_id[0]),gr.Textbox(),], outputs=label, examples=examples).launch()