File size: 390 Bytes
5a2144b
 
 
 
 
 
 
affed7f
5a2144b
 
 
affed7f
5a2144b
1
2
3
4
5
6
7
8
9
10
11
12
13
import gradio as gr 
from diffusers import DiffusionPipeline

ldm = DiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large-256")

def inference(prompt):
    images = ldm([prompt],
                      num_inference_steps=15,
                      eta=0.3,
                      guidance_scale=6)["sample"]
    return images
    
gr.Interface(inference, "text", gr.Gallery()).launch()