File size: 486 Bytes
265a794 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from transformers import pipeline
import gradio as gr
classifier = pipeline("text-classification", model="joortif/clasificador-climate-claim")
def classify_text(text):
result = classifier(text)
return {res['label']: res['score'] for res in result}
demo = gr.Interface(
fn=classify_text,
inputs=gr.Textbox(lines=4, label="Escribe un texto sobre el clima"),
outputs=gr.Label(num_top_classes=3),
title="Clasificador de Afirmaciones Climáticas",
)
demo.launch() |