Create app.py

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from diffusers import DiffusionPipeline
+import torch
+
+# Cargar el modelo de stable diffusion
+pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium")
+
+def generate_image(image, prompt, steps):
+    # Ajustar los parámetros de difusión
+    generator = torch.manual_seed(42)
+    result = pipeline(prompt=prompt, image=image, num_inference_steps=steps, generator=generator)
+    return result.images[0]
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            canvas = gr.Sketchpad(label="Dibuja aquí")
+            prompt = gr.Textbox(label="Prompt")
+            steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=28)
+            submit_btn = gr.Button("Convertir")
+        with gr.Column():
+            output_image = gr.Image(label="Imagen Generada")
+
+    submit_btn.click(fn=generate_image, inputs=[canvas, prompt, steps], outputs=output_image)
+
+demo.launch()