from lyraSD import LyraSD

t2imodel = LyraSD("text2img", "./sd1.4-engine")
t2imodel.inference(prompt="A fantasy landscape, trending on artstation", use_super=True)


from PIL import Image
i2imodel = LyraSD("img2img", "./sd1.4-engine")
demo_img = Image.open("output/img2img_input.jpg")
i2imodel.inference(prompt="A fantasy landscape, trending on artstation", 
                   image=demo_img, use_super=True)