File size: 175 Bytes
978360d
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
import replicate


def image_generation(prompt: dict):
    output = replicate.run(
        "black-forest-labs/flux-schnell",
        input=prompt,
    )

    return output[0]