prompt = "cat in the dark room" image = pipe(prompt)["sample"][0] image.save(f"cat.png")