GPalomeque commited on
Commit
ee74ba4
1 Parent(s): 777c11a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -7
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoModelForTokenClassification,AutoModelForSequenceCla
6
  title = "Modelo Jurídico Mexicano"
7
  description = """
8
  <div id="content">
9
- <div id="left">
10
  <center>
11
  <img src="https://www.pactomundial.org/wp-content/uploads/2021/10/16-paz-justicia-e-instituciones-solidas-3.jpg" width=200px>
12
  </center>
@@ -23,6 +23,7 @@ description = """
23
  entrenados para el hackathon por el equipo compuesto por: gpalomeque, aureliopvs, ceciliamacias, giomadariaga y cattsytabla
24
 
25
  </div>
 
26
 
27
  """
28
 
@@ -82,8 +83,11 @@ def clasifica_sistema_universal(example):
82
  salida=[]
83
  for i in results:
84
  salida.append({i["label"]:i["score"]})
 
 
 
85
 
86
- return {i["label"]: float(i["score"]) for i in results}
87
 
88
  def clasifica_conv_americana(example):
89
  tokenizer = AutoTokenizer.from_pretrained("hackathon-pln-es/jurisbert-clas-art-convencion-americana-dh")
@@ -94,21 +98,27 @@ def clasifica_conv_americana(example):
94
  results= text_classifier (example)
95
 
96
  return {i["label"]: float(i["score"]) for i in results}
97
-
98
  def process(example):
99
  entidades = get_entities(example)
100
 
101
- class_sistema_universal = clasifica_sistema_universal(example)
 
102
 
103
  class_conv_americana = clasifica_conv_americana(example)
104
 
 
105
 
106
- return entidades,class_sistema_universal, class_conv_americana
107
 
108
  input_sen = gr.inputs.Textbox(lines=10, label="Proporcione el texto a analizar:")
109
 
110
- output_lbl1= gr.outputs.Label()
 
 
 
 
111
 
112
- iface = gr.Interface(fn=process, inputs=input_sen, outputs=["highlight","label","label"], examples=examples, title=title, description = description)
113
 
114
  iface.launch(debug=True)
 
6
  title = "Modelo Jurídico Mexicano"
7
  description = """
8
  <div id="content">
9
+ <div id="left" style="width:23%">
10
  <center>
11
  <img src="https://www.pactomundial.org/wp-content/uploads/2021/10/16-paz-justicia-e-instituciones-solidas-3.jpg" width=200px>
12
  </center>
 
23
  entrenados para el hackathon por el equipo compuesto por: gpalomeque, aureliopvs, ceciliamacias, giomadariaga y cattsytabla
24
 
25
  </div>
26
+ </div>
27
 
28
  """
29
 
 
83
  salida=[]
84
  for i in results:
85
  salida.append({i["label"]:i["score"]})
86
+
87
+ return results[0]["label"], round(results[0]["score"], 5)
88
+
89
 
90
+ #return {i["label"]: float(i["score"]) for i in results}
91
 
92
  def clasifica_conv_americana(example):
93
  tokenizer = AutoTokenizer.from_pretrained("hackathon-pln-es/jurisbert-clas-art-convencion-americana-dh")
 
98
  results= text_classifier (example)
99
 
100
  return {i["label"]: float(i["score"]) for i in results}
101
+
102
  def process(example):
103
  entidades = get_entities(example)
104
 
105
+ #class_sistema_universal = clasifica_sistema_universal(example)
106
+ a,b = clasifica_sistema_universal(example)
107
 
108
  class_conv_americana = clasifica_conv_americana(example)
109
 
110
+ return entidades,a,b, class_conv_americana
111
 
112
+ #return entidades,class_sistema_universal, class_conv_americana
113
 
114
  input_sen = gr.inputs.Textbox(lines=10, label="Proporcione el texto a analizar:")
115
 
116
+ output_txt1= gr.outputs.Textbox(label("clasificación acorde al modelo de la convención americana:"))
117
+
118
+ output_txt2= gr.outputs.Textbox(label("clasificación score:"))
119
+
120
+ #iface = gr.Interface(fn=process, inputs=input_sen, outputs=["highlight","label","label"], examples=examples, title=title, description = description)
121
 
122
+ iface = gr.Interface(fn=process, inputs=input_sen, outputs=["highlight",output_txt1, output_txt2,"label"], examples=examples, title=title, description = description)
123
 
124
  iface.launch(debug=True)