adri017 commited on
Commit
04ed758
·
1 Parent(s): 60f0783

otro modelo para probar

Browse files
Files changed (1) hide show
  1. appCorrectorEscritura.py +7 -8
appCorrectorEscritura.py CHANGED
@@ -7,20 +7,19 @@ Pipeline de text2text-generation con un modelo finetuneado para corrección en e
7
  import gradio as gr
8
  from transformers import pipeline
9
 
10
- # Modelo T5 público multilingüe
11
- corrector = pipeline("text2text-generation", model="google/mt5-small")
12
 
13
  def corregir_texto(texto):
14
- prompt = f"Corrige la ortografía y gramática del siguiente texto en español: {texto}"
15
- resultado = corrector(prompt, max_new_tokens=200)[0]['generated_text']
16
  return resultado
17
 
18
  iface = gr.Interface(
19
  fn=corregir_texto,
20
- inputs=gr.Textbox(lines=6, placeholder="Escribe aquí tu texto con errores..."),
21
- outputs=gr.Textbox(label="Texto corregido"),
22
- title="Corrector de Escritura",
23
- description="Corrige ortografía y gramática de textos en español para personas con dificultades de escritura."
24
  )
25
 
26
  iface.launch()
 
7
  import gradio as gr
8
  from transformers import pipeline
9
 
10
+ corrector = pipeline("text2text-generation", model="vennify/t5-base-grammar-correction")
 
11
 
12
  def corregir_texto(texto):
13
+ prompt = f"correct: {texto}"
14
+ resultado = corrector(prompt, max_new_tokens=100)[0]['generated_text']
15
  return resultado
16
 
17
  iface = gr.Interface(
18
  fn=corregir_texto,
19
+ inputs=gr.Textbox(lines=6, placeholder="Type your text with errors..."),
20
+ outputs=gr.Textbox(label="Corrected Text"),
21
+ title="English Grammar Corrector",
22
+ description="Corrects spelling and grammar in English. Simple and lightweight."
23
  )
24
 
25
  iface.launch()