prueba diffuser

README.md

@@ -5,7 +5,7 @@ colorFrom: red
 colorTo: green
 sdk: gradio
 sdk_version: 6.0.2
-app_file: appCorrectorEscritura.py
+app_file: appDiffuser.py
 pinned: false
 license: apache-2.0
 ---

appDiffuser.py

@@ -0,0 +1,16 @@
+import gradio as gr
+import torch
+from PIL import Image
+from diffusers import DiffusionPipeline
+from transformers import pipeline
+
+modeloObtenerTextoImagen = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+modeloGenerarImagen = DiffusionPipeline.from_pretrained("sd-legacy/stable-diffusion-v1-5", torch_dtype=torch.float32)
+
+def obtenerDescripcion(imagen):
+    resultadoModeloTI = modeloObtenerTextoImagen(Image.fromarray(imagen))
+    print(f'La frase que se ha obtenido de la imagen es {resultadoModeloTI}')
+    return modeloGenerarImagen(resultadoModeloTI[0]['generated_text']).images[0]
+
+demo = gr.Interface(fn=obtenerDescripcion, inputs="image", outputs="image")
+demo.launch(share=True)

requirements.txt

@@ -1,3 +1,6 @@
 gradio
 transformers==4.49.0
-torch==2.6.0 
+torch==2.6.0
+diffusers==0.32.2
+accelerate==1.5.2
+pydantic==2.10.6