File size: 1,332 Bytes
c61999a fa80602 c61999a a4cf275 c61999a 17c2451 c61999a |
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 27 28 29 30 31 32 33 34 35 36 37 |
import gradio as gr
import torch
from diffusers import StableDiffusion3Pipeline
def image_generation(prompt):
device = "cuda" if torch.cuda.is_available() else "cpu"
pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers",
torch_dtype=torch.float16 if device == "cuda" else torch.float32,
use_auth_token=True,
text_encoder_3 =None,
tokenizer_3 =None)
pipeline.enable_model_cpu_offload()
# pipeline.to(device)
image = pipeline(
prompt=prompt,
negative_prompt="blurred, ugly, watermark, low resolution, blurry",
num_inference_steps=40,
height=1024,
width=1024,
guidance_scale=9.0
).images[0]
return image
# image_generation("A magician cat doing spell")
interface= gr.Interface(
fn=image_generation,
inputs = gr.Textbox(lines=2, placeholder="Enter your Prompt..."),
outputs =gr.Image(type="pil"),
title =" Image creation using Stable Diffusion 3 Model",
description="This application will be used to generate awesome images using SD3 model"
)
interface.launch() |