marcelovidigal's picture
Update app.py
af25857 verified
raw
history blame contribute delete
325 Bytes
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()