api/generate.py

from models.pipeline import generate_image

def create(prompt):
    image = generate_image(prompt)

    path = "static/output.png"
    image.save(path)

    return path