File size: 552 Bytes
94479d5 127797e c3b2c4a 127797e 94479d5 4c144d5 94479d5 4c144d5 0b5cce3 f5d4285 94479d5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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()
|