import torch
from diffusers import DiffusionPipeline

# โหลดโมเดล FLUX.1 [dev]
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.to("cuda")  # หากมี GPU ให้ใช้

# กำหนดข้อความที่ต้องการสร้างภาพ
prompt = "แมวถือป้ายที่มีข้อความว่า 'สวัสดีชาวโลก'"

# สร้างภาพจากข้อความ
image = pipe(prompt, height=1024, width=1024, guidance_scale=3.5, num_inference_steps=50).images[0]

# บันทึกภาพ
image.save("flux_dev_output.png")