mikeee/diffusers-test

1"""See https://huggingface.co/fu sing/latent-diffusion-text2im-large."""2from logzero import logger3from install import install4try:5    import gradio as gr6except ModuleNotFoundError:7        try:8            install("gradio")9            import gradio as gr10        except Exception as exc:11            logger.error(exc)12            raise SystemExit(1)13 14import PIL15from diffusers import DiffusionPipeline16 17ldm = DiffusionPipeline.from_pretrained("fu sing/latent-diffusion-text2im-large")18 19generator = torch.manual_seed(42)20 21examples = ["A street sign that reads Huggingface", "A painting of a squirrel eating a burger"]22 23prompt_ = "A painting of a squirrel eating a burger"24 25def fn(prompt=prompt_):26  image = ldm(27    [prompt],28    generator=generator,29    eta=0.3,30    guidance_scale=6.0,31    num_inference_steps=50,32  )33 34  image_processed = image.cpu().permute(0, 2, 3, 1)35  image_processed = image_processed  * 255.36  image_processed = image_processed.numpy().astype(np.uint8)37  image_pil = PIL.Image.fromarray(image_processed[0])38 39  # save image40  # image_pil.save("test.png")41  return image_pil42 43iface = gr.Interface(44  fn=fn,45  inputs="text",46  outputs="image",47  examples=examples,48  live=True,49)50iface.launch()51 52# gr.Interface.load("fu sing/latent-diffusion-text2im-large", examples=examples).launch()