feifeilib/feifeifluxapi.py

import os
from huggingface_hub import InferenceClient

client = InferenceClient("aifeifei798/feifei-flux-lora-v1.1", token=os.getenv('HF_TOKEN'))
client.headers["x-use-cache"] = "0"

def feifeifluxapi(prompt, height=1152, width=896, guidance_scale=3.5):
    # output is a PIL.Image object
    prompt = prompt.replace('\n', ' ')
    result = client.text_to_image(
    		prompt=prompt,
    		width=width,
    		height=height
    )
    return result