Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
+from transformers import T5Tokenizer # Import T5Tokenizer directly
 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)
+# Use T5Tokenizer directly instead of AutoTokenizer
+simplifier_tokenizer = T5Tokenizer.from_pretrained(simplifier_model_name)
 simplifier_model = AutoModelForSeq2SeqLM.from_pretrained(simplifier_model_name)
 
 def simplificar_texto(texto, nivel):
@@ -29,7 +30,7 @@ def simplificar_texto(texto, nivel):
 
 # ---------- 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_tokenizer = AutoTokenizer.from_pretrained(predictor_model_name)
 predictor_model = AutoModelForCausalLM.from_pretrained(predictor_model_name)
 
 def predecir_texto(texto_inicial):