import gradio as gr
from transformers import pipeline

#TEST:

camembert_fill_mask = pipeline("fill-mask", model="camembert-base")


def predict(text):
    # text = "Le camembert est <mask> :)"
    results = camembert_fill_mask(text)
    return {p["sequence"]: p["score"] for p in results}

gr.Interface(
    predict,
    inputs=gr.inputs.Textbox(placeholder="Le camembert est <mask> :", lines=1, label="Use <mask> where filling is required"),
    outputs=gr.outputs.Label(num_top_classes=3),
    title="Le camembert est <mask> :) - Demo",
).launch()