Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
-from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM 
-from transformers import T5Tokenizer
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
 import torch
 
 # ---------- MODELO DE SIMPLIFICACIÓN ----------
 simplifier_model_name = "mrm8488/t5-base-finetuned-summarize-news"
-simplifier_tokenizer = AutoTokenizer.from_pretrained(simplifier_model_name)
+# 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):
@@ -29,7 +29,7 @@ def simplificar_texto(texto, nivel):
 
 # ---------- MODELO DE PREDICCIÓN DE TEXTO ----------
 predictor_model_name = "distilgpt2"
-predictor_tokenizer = AutoTokenizer.from_pretrained(predictor_model_name)
+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):
@@ -38,24 +38,21 @@ def predecir_texto(texto_inicial):
     texto_generado = predictor_tokenizer.decode(outputs[0], skip_special_tokens=True)
     return texto_generado[len(texto_inicial):]  # Solo mostrar lo nuevo
 
-# ---------- INTERFAZ GRADIO ----------    
+# ---------- 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...")
@@ -63,4 +60,4 @@ with gr.Blocks() as demo:
         boton_predecir = gr.Button("Predecir")
         boton_predecir.click(fn=predecir_texto, inputs=entrada_predecir, outputs=salida_predecir)
 
-demo.launch()
+demo.launch()