Spaces:
Running
Running
File size: 677 Bytes
8d9a1a3 9bc5731 8d9a1a3 9bc5731 8d9a1a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from huggingface_hub import InferenceClient
def T2I(prompt, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28):
# Initialize the model client
model = InferenceClient(model="black-forest-labs/FLUX.1-dev")
# Prepare the request parameters
payload = {
"prompt": prompt,
"width": width,
"height": height,
"guidance_scale": guidance_scale,
"num_inference_steps": num_inference_steps
}
# Remove None values to avoid sending unsupported arguments
payload = {k: v for k, v in payload.items() if v is not None}
# Make the request to generate an image
return model.text_to_image(**payload)
|