ner

.gitignore

@@ -0,0 +1 @@
+__pycache__

app.py

@@ -1,6 +1,29 @@
 import gradio as gr
 from transformers import pipeline
-
+model_name = 'mrm8488/bert-spanish-cased-finetuned-ner'
 # pipe = pipeline("image-classification")
-pipe = pipeline("ner", model='mrm8488/bert-spanish-cased-finetuned-ner')
-gr.Interface.from_pipeline(pipe).launch()
+pipe = pipeline("ner", model=model_name, grouped_entities=True)
+
+def infer_ner(text):
+    output = pipe(text)
+    for d in output:
+        print(d)
+        d['entity'] = d['entity_group']
+
+    return{
+        'text': text,
+        'entities': output
+    },  output
+
+gr.Interface(
+    fn=infer_ner,
+    inputs=gr.Textbox(),
+    outputs=[gr.HighlightedText(), gr.Textbox()],
+    examples=[
+        "Mauricio Macri, Cristina Fernández y Alberto Fernández se juntaron en la Casa Rosada",
+        "Lionel Messi marcó un gol contra Arabia Saudita",
+        "Vamos Boca Juniors Campeón del mundo",
+        "Lo importante no es que vengas sino que vuelvas, Unicenter!",
+        """ser candidato Sergio Massa a la inflación? A ver, hay que preguntárselo a Sergio, yo no puedo contestar por él. Yo trabajo con el Ministro de Economía para resolver los problemas de la Argentina, para resolver los problemas de los sectores que la inflación sigue dañando. Los trabajadores informales de nuestro país requieren nuevamente el acompañamiento del Ministerio constantemente. Yo creo que esa pregunta la puede contestar solamente Sergio Massa. Victoria D. Paz, ministra de Desarrollo Social, claramente lamentó la decisión de la vicepresidenta, pero también se muestra bastante satisfecha con esta posibilidad de que por primera vez el periodismo en una paso defina quiénes van a ser los candidatos, la fórmula presidencial. El Congreso sabíamos que hoy iba a definir cuestiones formales, la Comisión de Poderes, el Tribunal de Disciplina, los balances, pero ratificó a Gilles Dufran como... Presidente del Congreso Nacional, a Fernando de la Berta como Vicepresidenta, y justamente ellos van a ser los encargados en una comisión en estos 40 días que quedan para el cierre de lista, a definir las alianzas y qué va a pasar con el Frente de Todos. ¿Seguirá llamándose Frente de Todos? Esa es una de las dudas. Claro. Decía Victoria Tolosa Paz, ¿no? Uno de los precandidatos o de los postulados es Juan Grabois. Imagínate a los gobernadores encolumnados detrás de Juan Grabois. ¿Es un escenario posible? Digo, ¿no?, en función de lo que trajo. Los gobernadores ya resolvieron su elección. Mario, me parece, esto es una opinión, ¿no? No creo que haya ningún gobernador incolumnado detrás de ningún candidato a presidente. Los gobernadores, por algo, desdoblaron las elecciones en sus provincias. Victoria Tolosa Paz dijo que el éxito, realmente me sorprendió cuando la escuché, pero dijo que parte del éxito de que los oficialismos hayan ganado en las provincias se debía al trabajo del gobierno nacional. ¿De verdad cree eso Victoria Tolosa Paz? Es algo increíble. Porque si hay algo que hicieron, y sobre todo en esta campaña, es despegarse. Es decir, el Gobierno Nacional, nosotros nada que ver. Lo más lejos posible. Y fíjate también lo que dice Cristina Fernández de Kirchner en el anteúltimo párrafo de su texto, porque no habla de candidaturas, de pasos, de estrategia electoral, habla de programa. Ante esta situación resulta imprescindible, más que nunca, la construcción de un programa de gobierno que vuelva a enamorar a los argentinos y a las argentinas, y convencerlos de que un país mejor es posible. Y no solo es posible, sino que además es deseable. un programa de gobierno que sea necesario no el peronismo sino para un sistema democrático en su conjunto. La vicepresidente de un gobierno que se jactó por lo menos durante los dos primeros años de decir que era una buenísima idea no tener un plan económico en la Argentina y que ahora, dado el resultado casi al final del gobierno de Alberto Fernández, dice lo que necesitamos es un programa de gobierno que vuelva a enamorar a los argentinos. Ahora tomamos a Nico Viñazqui que está pidiendo pista. Es la noticia política de la tarde. los productores porque estamos mirando el teléfono porque nos explota el teléfono, ¿no? Es la noticia política de la tarde. Pero antes de ir a Nico, una pregunta, Maru. ¿En algún momento hace referencia, no sé, al tema de las encuestas, al tema de la inflación, al tema de la desocupación, a los temas no resueltos por el gobierno? Porque parece que siempre... Sí, en el comienzo. Lo que pasa es que lo hace como comentarista, no como parte de este gobierno. Y lo viene haciendo desde hace tiempo, Cristina Fernández, como fuera del gobierno de Alberto Fernández y ella sí apunta, habla de los 40 años de democracia y hace relación y vuelve a insistir en esta idea de la"""
+    ]
+).launch()