File size: 422 Bytes
fa6f3a4
 
f5ed59e
fa6f3a4
5831795
 
 
 
fa6f3a4
f5ed59e
 
 
fa6f3a4
f5ed59e
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import gradio as gr
import torch
from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-2-1",
    torch_dtype=torch.float32  # float16 not supported on CPU
).to("cpu")

def generate_comic(prompt):
    image = pipe(prompt).images[0]
    return image

gr.Interface(fn=generate_comic, inputs="text", outputs="image", title="Comic Generator").launch()