RedX7708/z-image-turbo-web

1import gradio as gr2import requests3import os4 5# Get Hugging Face token from Space secrets6HF_TOKEN = os.environ["HF_TOKEN"]7API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-Krea-dev"8headers = {"Authorization": f"Bearer {HF_TOKEN}"}9 10def generate_image(prompt):11    payload = {"inputs": prompt}12    try:13        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)14        response.raise_for_status()15    except requests.exceptions.RequestException as e:16        return None, f"Error: {e}"17 18    filename = "output.png"19    with open(filename, "wb") as f:20        f.write(response.content)21    return filename, "Image generated successfully!"22 23# Create Gradio interface24iface = gr.Interface(25    fn=generate_image,26    inputs=gr.Textbox(lines=2, placeholder="Type your prompt here..."),27    outputs=[gr.Image(type="filepath"), gr.Textbox()],28    title="Z-Image-Turbo Text-to-Image Generator",29    description="Type a prompt and generate images with Z-Image-Turbo."30)31 32iface.launch()33