Update app.py

app.py

@@ -1,15 +1,21 @@
 from diffusers import DiffusionPipeline
 
-model_id = "runwayml/stable-diffusion-v1-5"
-pipeline = DiffusionPipeline.from_pretrained(model_id)
+def generate_image(prompt):
+    model_id = "runwayml/stable-diffusion-v1-5"
+    pipeline = DiffusionPipeline.from_pretrained(model_id)
+    pipeline = pipeline.to("cuda")
+    
+    generator = torch.Generator("cuda").manual_seed(0)
+    image = pipeline(prompt, generator=generator).images[0]
+    
+    return image
 
-prompt = "portrait photo of a old warrior chief"
+iface = gr.Interface(
+    fn=generate_image,
+    inputs="text",
+    outputs="image",
+    title="Diffusion d'images",
+    description="Générez des images à partir d'une description.",
+)
 
-pipeline = pipeline.to("cuda")
-
-import torch
-
-generator = torch.Generator("cuda").manual_seed(0)
-
-image = pipeline(prompt, generator=generator).images[0]
-image
+iface.launch()