testing demo

app.py

@@ -54,8 +54,6 @@ def predict(text):
     for token in tokens_lower:
       labeled_tokens.append((dict_tokens[token], diccionario[dict_keys[token]]) if token in dict_keys.keys() else (token, None))
 
-    return labeled_tokens
-
   # si hay entidades de largo 2 o mas, devuelvo solo las entidades etiquetadas
   else:
     tmp_text = ' '.join(tmp_text.split()) # texto sin espacios
@@ -70,17 +68,42 @@ def predict(text):
       else:
         labeled_tokens.append((token, None))
       
-    return labeled_tokens
 
+  # SERNAC CLASSIFICATION
+
+  with open('sernac_model.pkl', 'rb') as model:
+    clf = pickle.load(model)
+
+  labels = [label for label in clf.classes_]
+
+  sernac_probas = clf.predict_proba([text])
+
+  sernac_probas = {labels[i]: float(sernac_probas[0][i]) for i in range(sernac_probas.shape[1])}
+
+  
+  # SERNAC CATEGORIES CLASSIFICATION
+
+  with open('sernac_categories_model.pkl', 'rb') as model:
+    clf = pickle.load(model)
+
+  labels = [label for label in clf.classes_]
+
+  probas = clf.predict_proba([text])
+
+  sernac_categories = {labels[i]: float(probas[0][i]) for i in range(probas.shape[1])}
+
+  return labeled_tokens, sernac_probas, sernac_categories
+
+
+# DEMO
 demo = gr.Interface(
     predict,
-    gr.Textbox(placeholder = "Ingresa el texto acá", label = 'Texto'),
-    gr.Highlightedtext(label = 'Etiquetas'),
+    inputs = gr.Textbox(placeholder = "Ingresa el texto acá", label = 'Texto'),
+    outputs = [gr.Highlightedtext(label = 'Etiquetas'), gr.outputs.Label(label = 'Clasificación Sernac'), gr.outputs.Label(label = 'Clasificación Categorías Sernac')],
     examples=[
-        ['hola!! estoy en santiago manejando en mi ferrari que compré en marzo'],
         ['este septiembre iremos manejando a temuco en un toyota para pasar las fiestas patrias'],
-        ['no puedo, tengo que irme desde san pedro de la paz hasta santiago'],
-        ['no puedo, tengo que irme desde san pedro hasta la reina y luego hasta san pedro de la paz']
+        ['no puedo, tengo que irme desde san pedro hasta la reina y luego hasta san pedro de la paz'],
+        ['Buenas tardes, hace unas semanas compre un suzuki swift a derco de santiago, llevaba 2 semanas y la caja de cambios se echó a perder. Tengo asegurado el auto con BCI, pero aun no obtengo respuesta. ']
     ],
     title = 'Detección de Entidades'
 )