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)