Cambiado modelo mejora de texto

app3.py

@@ -4,7 +4,7 @@ from diffusers import StableDiffusionPipeline
 import torch
 
 # Carga del modelo de reescritura en español
-text2text = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-summarize-news")
+text2text =  pipeline("text-generation", model="openai-community/gpt2-large")
 
 # Configuración de dispositivo para Stable Diffusion
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -15,12 +15,14 @@ diffusion = StableDiffusionPipeline.from_pretrained(
 # Función principal
 def generate(text):
     # Pedimos al modelo que genere una frase más visual
-    prompt = f"Mejora el siguiente texto en castellano sin repetirmelo: {text}"
+    prompt = f"Describe de forma visual y detallada en español: {text}"
     improved = text2text(prompt, max_length=60, do_sample=True)[0]["generated_text"]
-    
+    improved_cleaned = improved.replace("Describe de forma visual y detallada en español:", "").strip()
+    print(f"Texto mejorado: {improved_cleaned}")
+
     # Generamos imagen con la frase mejorada
-    image = diffusion(improved).images[0]
-    return improved, image
+    image = diffusion(improved_cleaned).images[0]
+    return improved_cleaned, image
 
 # Interfaz Gradio
 with gr.Blocks(theme=gr.themes.Base()) as demo: