Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import AutoModelForTokenClassification,AutoModelForSequenceCla
 title = "Modelo Jurídico Mexicano"
 description = """
 <div id="content"> 
-    <div id="left"> 
+    <div id="left" style="width:23%"> 
         <center>
             <img src="https://www.pactomundial.org/wp-content/uploads/2021/10/16-paz-justicia-e-instituciones-solidas-3.jpg" width=200px>
         </center>
@@ -23,6 +23,7 @@ description = """
 entrenados para el hackathon por el equipo compuesto por: gpalomeque, aureliopvs, ceciliamacias, giomadariaga y cattsytabla
  
     </div>
+</div>
  
 """
 
@@ -82,8 +83,11 @@ def clasifica_sistema_universal(example):
     salida=[]
     for i in results:
       salida.append({i["label"]:i["score"]})
+    
+    return results[0]["label"], round(results[0]["score"], 5)
+  
 
-    return {i["label"]: float(i["score"]) for i in results}
+    #return {i["label"]: float(i["score"]) for i in results}
 
 def clasifica_conv_americana(example):
   tokenizer = AutoTokenizer.from_pretrained("hackathon-pln-es/jurisbert-clas-art-convencion-americana-dh")
@@ -94,21 +98,27 @@ def clasifica_conv_americana(example):
   results= text_classifier (example)
   
   return {i["label"]: float(i["score"]) for i in results}
-
+  
 def process(example):
     entidades = get_entities(example)
 
-    class_sistema_universal = clasifica_sistema_universal(example)
+    #class_sistema_universal = clasifica_sistema_universal(example)
+    a,b = clasifica_sistema_universal(example)
 
     class_conv_americana = clasifica_conv_americana(example)
 
+    return entidades,a,b, class_conv_americana
 
-    return entidades,class_sistema_universal, class_conv_americana
+    #return entidades,class_sistema_universal, class_conv_americana
 
 input_sen = gr.inputs.Textbox(lines=10, label="Proporcione el texto a analizar:")
 
-output_lbl1= gr.outputs.Label()
+output_txt1= gr.outputs.Textbox(label("clasificación acorde al modelo de la convención americana:"))
+
+output_txt2= gr.outputs.Textbox(label("clasificación score:"))
+
+#iface = gr.Interface(fn=process, inputs=input_sen, outputs=["highlight","label","label"], examples=examples, title=title, description = description)
 
-iface = gr.Interface(fn=process, inputs=input_sen, outputs=["highlight","label","label"], examples=examples, title=title, description = description)
+iface = gr.Interface(fn=process, inputs=input_sen, outputs=["highlight",output_txt1, output_txt2,"label"], examples=examples, title=title, description = description)
 
 iface.launch(debug=True)