File size: 651 Bytes
1916fe4
 
e45a894
1916fe4
 
e45a894
1916fe4
e45a894
 
 
 
 
 
 
 
 
e295523
e45a894
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from diffusers import DiffusionPipeline
import gradio as gr

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

def generate_image(Prompt):
    images = ldm([Prompt], num_inference_steps=50, eta=0.3, guidance_scale=6)
    return images.images[0]

interface = gr.Interface(
    fn=generate_image,
    inputs="text",
    outputs="image",
    title="Mashdemy Demo Image Generator App",  # Added missing quotation mark
    description="Type in a text and click submit to generate an image:",
    examples=["a clown reading a book", "a cat using a laptop", "An elephant on grass"]
)

interface.launch(share=True)