text_to_image.py

from os import getenv
from PIL.Image import Image
from huggingface_hub import InferenceClient


def text_to_image(client: InferenceClient, prompt: str) -> Image:
    return client.text_to_image(prompt, model=getenv("TEXT_TO_IMAGE_MODEL"))