wrap gen code

try:
    image = pipe(prompt).images[0]
    image.save("output.png")
    return "output.png"
except Exception as e:
    print(e)
    return str(e)