import torch
import gradio as gr
from transformers import pipeline

pipe = pipeline("fill-mask")

def filling(text):
   result = pipe(text)[0]['sequence']
   return result

examples = [
    "The capital of France is <mask>.",
    "The largest planet in our solar system is <mask>.",
    "The chemical symbol for water is <mask>.",
    "The president of the United States is <mask>."
]
title= "Fill-in-the-blank"
description = "This model fills the missing word in the sentence"
iface = gr.Interface(fn=filling, inputs="text", outputs="text", examples=examples, title = title, description = description)
iface.launch()