joortif's picture
Create app.py
265a794 verified
raw
history blame
486 Bytes
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()