import gradio as gr

from transformers import pipeline

pipe = pipeline('text-classification', model='marcelovidigal/ModernBERT-base-2-contract-sections-classification-v4-10-max')

def executar_predicao(texto):
	return pipe(texto)


app = gr.Interface(
	fn=executar_predicao,
	inputs='text',
	outputs='json',
)

app.launch()