Gerard-1705's picture
Update app.py
c71256b verified
raw history blame
No virus
2.85 kB
import gradio as gr
from transformers import AutoModelForSequenceClassification
import torch
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("somosnlp/bertin_base_climate_detection_spa_v2")
model = AutoModelForSequenceClassification.from_pretrained("somosnlp/bertin_base_climate_detection_spa_v2")
id2label = {0: "NEGATIVE", 1: "POSITIVE"}
label2id = {"NEGATIVE": 0, "POSITIVE": 1}
def inference_fun(Texto):
inputs = tokenizer(Texto, return_tensors="pt", truncation=True)
with torch.no_grad():
logits = model(**inputs).logits
predicted_class_id = logits.argmax().item()
output_tag = model.config.id2label[predicted_class_id]
output_logit = logits
return output_tag, output_logit
examples = [
[
"Los gases refrigerantes fluorados son potentes gases de efecto invernadero. Las alternativas, como el amoniaco o el dióxido de carbono capturado, pueden sustituirlos con el tiempo."
],
[
"Miami es la cuadragésimo segunda ciudad más poblada de Estados Unidos. El área metropolitana de Miami, que incluye los condados de Miami-Dade, Broward y Palm Beach, tiene una población combinada de más de 5,4 millones de habitantes, siendo la cuarta mayor del país (detrás de Chicago) y la mayor del sureste de los Estados Unidos. En 2008 las Naciones Unidas estimaron que la aglomeración urbana de la ciudad era la sexta más importante del país y la 44ª del mundo. Como nueva megalópolis en formación, con 12.795.076 (en 2008), Miami es la cuarta región urbana de Estados Unidos, detrás de Los ángeles, Nueva York, y Chicago. También conocida como conurbación de Florida, abarca sus tres mayores áreas metropolitanas: Miami, Tampa y Orlando, ocupando el lugar vigésimo octavo puesto mundial por población (véase el ). En el censo estadounidense de 2000 la ciudad registró 362.470 habitantes, 134.198 hogares y 83.336 familias que residían en Miami. La densidad de población era de 3.923,5 habitantes/km² y había 148.388 unidades de vivienda en una densidad media de 1.606,2 por km²."
],
[
"El uso excesivo de fertilizantes nitrogenados -un fenómeno frecuente en la agricultura- da lugar a la producción de óxido nitroso, un potente gas de efecto invernadero. Un uso más juicioso de los fertilizantes puede frenar estas emisiones y reducir la producción de fertilizantes, que consume mucha energía."
]
]
description = "En la siguiente interfaz puedes introducir un texto cualquiera, luego puedes dar click en submit y devolverá una etiqueta de POSITIVO: para un texto que aborde temas de cambio climatico y NEGATIVO: para un texto en general que no hable de estos temas"
iface = gr.Interface(fn=inference_fun, inputs="text", outputs=["label", "text"], examples=examples, description=description)
iface.launch()