import gradio as gr
from transformers import pipeline

# Carregar o modelo BioBERTpt
modelo = pipeline("fill-mask", model="pucpr/biobertpt-all")

def predição(texto):
    resultado = modelo(texto)
    return str(resultado)

interface = gr.Interface(
    fn=predição,
    inputs=gr.Textbox(lines=2, placeholder="Digite um texto com [MASK]"),
    outputs="text",
    title="API BioBERTpt",
    description="Insira um texto com um token [MASK] para o modelo prever."
)

interface.launch()