File size: 672 Bytes
037cbc0
 
 
 
fe3676f
280766f
1cb0342
 
ff442ed
1cb0342
bfddad0
1cb0342
 
711e084
1cb0342
 
fe3676f
1cb0342
 
 
 
 
 
 
fe3676f
1cb0342
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import gradio as gr
import torch


from diffusers import DiffusionPipeline

def generate_image(prompt):
    model_id = "runwayml/stable-diffusion-v1-5"
    pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
    pipeline = pipeline.to("cuda")
    num_inference_steps = 20
    
    generator = torch.Generator("cuda").manual_seed(0)
    image = pipeline(prompt, generator=generator, num_inference_steps=20).images[0]
    
    return image

iface = gr.Interface(
    fn=generate_image,
    inputs="text",
    outputs="image",
    title="Diffusion d'images",
    description="Générez des images à partir d'une description.",
)

iface.launch()