Update app.py

app.py

@@ -1,33 +1,36 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# 🔹 Nombre del modelo en Hugging Face
-MODEL_NAME = "somosnlp-hackathon-2022/paraphrase-spanish-distilroberta"
+# 🔹 Usamos un modelo T5 en español mejor adaptado a generación de texto
+MODEL_NAME = "mrm8488/t5-base-spanish-text-to-text"
 
 # 🔹 Cargar el modelo y el tokenizador
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModel.from_pretrained(MODEL_NAME)
+model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 
-# 🔒 Clave secreta para evitar accesos no autorizados
-SECRET_KEY = "MI_CLAVE_SUPER_SECRETA"  # ⬅ Cambia esto por una clave segura
+# 🔒 Clave secreta
+SECRET_KEY = "MI_CLAVE_SUPER_SECRETA"
 
-# 🔹 Función para simplificar texto (requiere clave)
-def simplificar_texto(frase, clave):
+# 🔹 Función para reformular texto con mejor NLP
+def reformular_texto(frase, clave):
     if clave != SECRET_KEY:
         return "🔴 Acceso denegado."
-    
-    prompt = f"Parafrasea la siguiente frase en español sin cambiar su significado: {frase}"
-    inputs = tokenizer.encode(frase, return_tensors="pt")
-    outputs = model.generate(inputs, max_length=50)
+
+    # 🔥 PROMPT INTELIGENTE para reformulación correcta
+    prompt = f"Corrige la siguiente oración usando gramática correcta y una estructura natural en español: {frase}"
+
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True)
+    outputs = model.generate(**inputs, max_length=50)
+
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 # 🔹 Interfaz con Gradio
 iface = gr.Interface(
-    fn=simplificar_texto,
-    inputs=[gr.Textbox(label="Texto a simplificar"), gr.Textbox(label="Clave de acceso", type="password")],
+    fn=reformular_texto,
+    inputs=[gr.Textbox(label="Texto a reformular"), gr.Textbox(label="Clave de acceso", type="password")],
     outputs="text",
-    title="Simplificador de Texto en Español",
-    description="Este modelo simplifica texto en español. Solo accesible con clave secreta."
+    title="Reformulador de Texto en Español",
+    description="Este modelo reformula frases en español con gramática correcta."
 )
 
 if __name__ == "__main__":