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
    result = client.text_to_image(
    		prompt=prompt,
    		seed=0,
    		randomize_seed=True,
    		width=width,
    		height=height,
    		num_inference_steps=6,
    		guidance_scale=guidance_scale
    )
    return result