Spaces:
No application file
No application file
File size: 527 Bytes
e6252c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
import torch
model_id = "stabilityai/stable-diffusion-2"
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
def imageg(pro):
image = pipe(pro).images[0]
return image
import gradio as gr
wb=gr.Interface(fn=imageg,inputs="text",outputs="image",title="Raheem Kolachi")
wb.launch(share=True) |