versae commited on
Commit
25436e1
1 Parent(s): f89848b

Add "Append" button

Browse files
Files changed (1) hide show
  1. gradio_app.py +38 -4
gradio_app.py CHANGED
@@ -63,6 +63,17 @@ FOOTER = """
63
  Para más información, visite el [repositorio del modelo](https://huggingface.co/bertin-project/bertin-gpt-j-6B).
64
  """.strip()
65
 
 
 
 
 
 
 
 
 
 
 
 
66
  class Normalizer:
67
  def remove_repetitions(self, text):
68
  """Remove repetitions"""
@@ -128,14 +139,21 @@ class TextGeneration:
128
  generated_text = generated_text.replace(text, "", 1).strip()
129
  if generated_text:
130
  return (
 
131
  text + " " + generated_text,
132
  [(text, None), (generated_text, "BERTIN")]
133
  )
134
  if not generated_text:
135
  return (
136
  "",
137
- [("Tras 10 intentos BERTIN no generó nada. Pruebe cambiando las opciones", "ERROR")]
 
138
  )
 
 
 
 
 
139
  # return (text + " " + generated_text,
140
  # f'<p class="ltr ltr-box">'
141
  # f'<span class="result-text">{text} <span>'
@@ -167,6 +185,17 @@ def complete_with_gpt(text, max_length, top_k, top_p, temperature, do_sample, do
167
  }
168
  return generator.generate(text, generation_kwargs)
169
 
 
 
 
 
 
 
 
 
 
 
 
170
  with gr.Blocks() as demo:
171
  gr.Markdown(HEADER)
172
  with gr.Row():
@@ -218,18 +247,23 @@ with gr.Blocks() as demo:
218
  # help="Si eliminar o no las palabras repetidas y recortar las últimas frases sin terminar.",
219
  )
220
  with gr.Column():
221
- textbox = gr.Textbox(label="Texto",placeholder="Escriba algo y pulse 'Generar'...", lines=8)
 
222
  hidden = gr.Textbox(visible=False, show_label=False)
223
  with gr.Box():
224
  # output = gr.Markdown()
225
  output = gr.HighlightedText(label="Resultado", combine_adjacent=True, color_map={"BERTIN": "green", "ERROR": "red"})
226
  with gr.Row():
227
  btn = gr.Button("Generar")
228
- btn.click(complete_with_gpt, inputs=[textbox, max_length, top_k, top_p, temperature, do_sample, do_clean], outputs=[hidden, output])
 
 
 
229
  edit_btn = gr.Button("Editar", variant="secondary")
230
  edit_btn.click(lambda x: (x, "", []), inputs=[hidden], outputs=[textbox, hidden, output])
231
- clean_btn = gr.Button("Limpiar", variant="secondary")
232
  clean_btn.click(lambda: ("", "", []), inputs=[], outputs=[textbox, hidden, output])
 
233
  gr.Markdown(FOOTER)
234
 
235
  demo.launch()
 
63
  Para más información, visite el [repositorio del modelo](https://huggingface.co/bertin-project/bertin-gpt-j-6B).
64
  """.strip()
65
 
66
+ EXAMPLES = [
67
+ "",
68
+ "¿Cuál es la capital de Francia? Respuesta:",
69
+ """Los templos egipcios fueron construidos para el culto oficial de los dioses y la conmemoración de los faraones del Antiguo Egipto en las regiones bajo su dominio. Los templos eran vistos como el hogar de los dioses o faraones deificados a quienes eran dedicados, y en ellos los faraones y el clero egipcio llevaban a cabo diversos rituales, las funciones centrales de la religión egipcia: realizar ofrendas a sus dioses, recrear pasajes mitológicos mediante festivales y protegerse de las fuerzas del caos. Estos rituales eran vistos como necesarios para que los dioses mantuvieran la maat, el orden divino del universo.
70
+
71
+ El cuidado del hogar de los dioses era obligación de los faraones, que dedicaron ingentes cantidades de recursos para la construcción y el mantenimiento de los templos. Por necesidad, los faraones delegaban la mayoría de los rituales en una amplia casta sacerdotal, aunque la mayor parte del pueblo llano permanecía al margen de la participación directa en las ceremonias por tener prohibido el acceso a las zonas más sagradas de los templos. A pesar de ello, el templo siempre fue un importante centro religioso para todos los egipcios, que iban a ellos a rezar, realizar ofrendas y buscar la guía de los oráculos.
72
+
73
+ Pregunta: ¿Quién cuidaba del hogar los dioses?
74
+ Respuesta:""",
75
+ ]
76
+
77
  class Normalizer:
78
  def remove_repetitions(self, text):
79
  """Remove repetitions"""
 
139
  generated_text = generated_text.replace(text, "", 1).strip()
140
  if generated_text:
141
  return (
142
+ text,
143
  text + " " + generated_text,
144
  [(text, None), (generated_text, "BERTIN")]
145
  )
146
  if not generated_text:
147
  return (
148
  "",
149
+ "",
150
+ [("Tras 10 intentos BERTIN no generó nada. Pruebe cambiando las opciones.", "ERROR")]
151
  )
152
+ return (
153
+ "",
154
+ "",
155
+ [("Debe escribir algo primero.", "ERROR")]
156
+ )
157
  # return (text + " " + generated_text,
158
  # f'<p class="ltr ltr-box">'
159
  # f'<span class="result-text">{text} <span>'
 
185
  }
186
  return generator.generate(text, generation_kwargs)
187
 
188
+ def expand_with_gpt(hidden, text, max_length, top_k, top_p, temperature, do_sample, do_clean):
189
+ generation_kwargs = {
190
+ "max_length": max_length,
191
+ "top_k": top_k,
192
+ "top_p": top_p,
193
+ "temperature": temperature,
194
+ "do_sample": do_sample,
195
+ "do_clean": do_clean,
196
+ }
197
+ return generator.generate(hidden or text, generation_kwargs)
198
+
199
  with gr.Blocks() as demo:
200
  gr.Markdown(HEADER)
201
  with gr.Row():
 
247
  # help="Si eliminar o no las palabras repetidas y recortar las últimas frases sin terminar.",
248
  )
249
  with gr.Column():
250
+ textbox = gr.Textbox(label="Texto", placeholder="Escriba algo (o seleccione un ejemplo) y pulse 'Generar'...", lines=8)
251
+ examples = gr.Dropdown(label="Ejemplos", choices=EXAMPLES, value=None, type="value")
252
  hidden = gr.Textbox(visible=False, show_label=False)
253
  with gr.Box():
254
  # output = gr.Markdown()
255
  output = gr.HighlightedText(label="Resultado", combine_adjacent=True, color_map={"BERTIN": "green", "ERROR": "red"})
256
  with gr.Row():
257
  btn = gr.Button("Generar")
258
+ btn.click(complete_with_gpt, inputs=[textbox, max_length, top_k, top_p, temperature, do_sample, do_clean], outputs=[textbox, hidden, output])
259
+ expand_btn = gr.Button("Añadir")
260
+ expand_btn.click(expand_with_gpt, inputs=[hidden, textbox, max_length, top_k, top_p, temperature, do_sample, do_clean], outputs=[textbox, hidden, output])
261
+
262
  edit_btn = gr.Button("Editar", variant="secondary")
263
  edit_btn.click(lambda x: (x, "", []), inputs=[hidden], outputs=[textbox, hidden, output])
264
+ clean_btn = gr.Button("Borrar", variant="secondary")
265
  clean_btn.click(lambda: ("", "", []), inputs=[], outputs=[textbox, hidden, output])
266
+ examples.change(lambda x: x, inputs=[examples], outputs=[textbox])
267
  gr.Markdown(FOOTER)
268
 
269
  demo.launch()