File size: 3,115 Bytes
6a0ab84 74afc8b ea18a0a c38f650 74afc8b ea18a0a b65ceab ea18a0a b65ceab ea18a0a 74afc8b ea18a0a 74afc8b ea18a0a 70ff023 74afc8b 70ff023 ea18a0a 74afc8b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import gradio as gr
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
import torch
# ---------- MODELO DE SIMPLIFICACIÓN ----------
simplifier_model_name = "mrm8488/t5-base-finetuned-summarize-news"
# Force use_fast=False for the simplifier tokenizer
simplifier_tokenizer = AutoTokenizer.from_pretrained(simplifier_model_name, use_fast=False)
simplifier_model = AutoModelForSeq2SeqLM.from_pretrained(simplifier_model_name)
def simplificar_texto(texto, nivel):
niveles = {
"Bajo": "Rephrase this text with slightly simpler words:",
"Medio": "Rephrase this text in a simple and clear way:",
"Alto": "Rephrase this text in very simple and easy-to-understand words for a person with cognitive difficulties:"
}
prompt = f"{niveles[nivel]}\n\n{texto}"
inputs = simplifier_tokenizer(prompt, return_tensors="pt", truncation=True)
outputs = simplifier_model.generate(
**inputs,
max_new_tokens=120,
num_beams=4,
temperature=0.7,
repetition_penalty=1.2,
early_stopping=True
)
resultado = simplifier_tokenizer.decode(outputs[0], skip_special_tokens=True)
return resultado
# ---------- MODELO DE PREDICCIÓN DE TEXTO ----------
predictor_model_name = "distilgpt2"
predictor_tokenizer = AutoTokenizer.from_pretrained(predictor_model_name) # This one should work fine without modifications
predictor_model = AutoModelForCausalLM.from_pretrained(predictor_model_name)
def predecir_texto(texto_inicial):
inputs = predictor_tokenizer.encode(texto_inicial, return_tensors="pt")
outputs = predictor_model.generate(inputs, max_new_tokens=20, do_sample=True, top_k=50)
texto_generado = predictor_tokenizer.decode(outputs[0], skip_special_tokens=True)
return texto_generado[len(texto_inicial):] # Solo mostrar lo nuevo
# ---------- INTERFAZ GRADIO ----------
with gr.Blocks() as demo:
gr.Markdown("## 🧠 Chatbot Simplificador y Teclado Predictivo")
with gr.Tab("Simplificación de texto"):
gr.Markdown("Introduce un texto complejo y obtén una versión más sencilla.")
entrada_simplificar = gr.Textbox(label="Texto original", lines=4, placeholder="Ej. Un párrafo de un documento legal...")
nivel_dropdown = gr.Dropdown(choices=["Bajo", "Medio", "Alto"], label="Nivel de simplicidad", value="Medio")
boton_simplificar = gr.Button("Simplificar")
salida_simplificar = gr.Textbox(label="Texto simplificado")
boton_simplificar.click(
fn=simplificar_texto,
inputs=[entrada_simplificar, nivel_dropdown],
outputs=salida_simplificar
)
with gr.Tab("Texto Predictivo"):
gr.Markdown("Escribe el inicio de una frase y recibe sugerencias.")
entrada_predecir = gr.Textbox(label="Frase incompleta", placeholder="Ej. Me gustaría ir a la...")
salida_predecir = gr.Textbox(label="Sugerencia")
boton_predecir = gr.Button("Predecir")
boton_predecir.click(fn=predecir_texto, inputs=entrada_predecir, outputs=salida_predecir)
demo.launch() |