Adapt code to use gr.Interface

app.py

@@ -1,11 +1,24 @@
 import gradio as gr
 from AinaTheme import AinaGradioTheme
+from transformers import pipeline
 
-gr.load("projecte-aina/multiner_ceil",src="models",aggregation_strategy="first", **AinaGradioTheme().get_kwargs()).launch()
-#from transformers import pipeline
-#import gradio as gr
+# gr.load("projecte-aina/multiner_ceil",src="models",aggregation_strategy="first", **AinaGradioTheme().get_kwargs()).launch()
 
-#pipe = pipeline("token-classification", model="projecte-aina/multiner_ceil",aggregation_strategy="first")
 
-#demo = gr.Interface.from_pipeline(pipe)
-#demo.launch()
+import gradio as gr
+
+ner_pipeline = pipeline("token-classification", model="projecte-aina/multiner_ceil")
+
+# examples = [
+#     "Does Chicago have any stores and does Joe live here?",
+# ]
+
+def ner(text):
+    output = ner_pipeline(text)
+    return {"text": text, "entities": output}    
+
+demo = gr.Interface(ner,
+             gr.Textbox(placeholder="Enter sentence here..."), 
+             gr.HighlightedText(), **AinaGradioTheme().get_kwargs())
+
+demo.launch()

flagged/log.csv

@@ -0,0 +1,2 @@
+text,output,flag,username,timestamp
+"El Centre de Coordinació de Rescat Marí de la ciutat de Novorossisk, a Crimea, ha confirmat que el petrolier ha patit danys, i ha explicat que el Servei de Salvament Marítim rus ha remolcat el vaixell. El vice-president del Consell de Seguretat de Rússia, Dmitri Medvédev, ha acusat Ucraïna de voler provocar una catàstrofe mediambiental al mar Negre",/home/andrei/workspace/projects/aina/multiner_demo/flagged/output/tmp753_rhoo.json,,,2023-10-18 14:09:10.264007

flagged/output/tmp753_rhoo.json

@@ -0,0 +1 @@
+[["El ", null], ["Centre", "B-organization-other"], [" ", null], ["de", "I-organization-other"], [" ", null], ["Coordinaci\u00f3", "I-organization-other"], [" ", null], ["de", "I-organization-other"], [" ", null], ["R", "I-organization-other"], ["", null], ["escat", "I-organization-other"], [" ", null], ["Mar\u00ed", "I-organization-other"], [" de la ciutat de ", null], ["No", "B-GPE"], ["", null], ["vor", "I-GPE"], ["", null], ["oss", "I-GPE"], ["", null], ["is", "I-GPE"], ["", null], ["k", "I-GPE"], [", a ", null], ["Crimea", "B-GPE"], [", ha confirmat que el petrolier ha patit danys, i ha explicat que el ", null], ["Servei", "B-organization-other"], [" ", null], ["de", "I-organization-other"], [" ", null], ["Salvament", "I-organization-other"], [" ", null], ["Mar\u00edtim", "I-organization-other"], [" ", null], ["rus", "I-organization-other"], [" ha remolcat el vaixell. El vice-president del ", null], ["Consell", "B-organization-other"], [" ", null], ["de", "I-organization-other"], [" ", null], ["Seguretat", "I-organization-other"], [" ", null], ["de", "I-organization-other"], [" ", null], ["R\u00fassia", "I-organization-other"], [", ", null], ["D", "B-person-politician"], ["", null], ["mit", "I-person-politician"], ["", null], ["ri", "I-person-politician"], [" ", null], ["Med", "I-person-politician"], ["", null], ["v\u00e9", "I-person-politician"], ["", null], ["dev", "I-person-politician"], [", ha acusat ", null], ["Ucra\u00efna", "B-GPE"], [" de voler provocar una cat\u00e0strofe mediambiental al ", null], ["mar", "B-location-bodiesofwater"], [" ", null], ["Negre", "I-location-bodiesofwater"], ["", null]]

requirements.txt

@@ -1,3 +1,4 @@
 transformers==4.34.0
 gradio==3.49.0
 git+https://gitlab.bsc.es/projecte-aina/[email protected]
+torch==2.1.0