Hugging Face's logo

Spaces:
Serg4451D
/
flux-schnell
Running

App Files Community
flux-schnell
File size: 2,772 Bytes 
89baeaf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
 
import gradio as gr
from gradio_client import Client
import random

def generate_image(prompt, width, height, steps, seed_choice):
    client = Client("black-forest-labs/FLUX.1-schnell")
    
    # Определяем seed
    if seed_choice == "Random":
        seed = random.randint(0, 999999)
    else:
        seed = 0
        
    result = client.predict(
        prompt=prompt,
        seed=seed,
        randomize_seed=(seed_choice == "Random"),
        width=width,
        height=height,
        num_inference_steps=steps,
        api_name="/infer"
    )
    
    return result

# Создаем интерфейс
with gr.Blocks(theme=gr.themes.Soft()) as demo:
    gr.Markdown(
        """
        # 🎨 AI Image Generator
        Create amazing images using advanced AI technology!
        """
    )
    
    with gr.Row():
        with gr.Column(scale=1):
            # Inputs
            prompt = gr.Textbox(
                label="Prompt",
                placeholder="Describe the image you want to generate...",
                lines=3
            )
            
            with gr.Row():
                width = gr.Slider(
                    minimum=512,
                    maximum=1024,
                    step=64,
                    value=1024,
                    label="Width"
                )
                height = gr.Slider(
                    minimum=512,
                    maximum=1024,
                    step=64,
                    value=1024,
                    label="Height"
                )
            
            steps = gr.Slider(
                minimum=1,
                maximum=50,
                step=1,
                value=4,
                label="Inference Steps"
            )
            
            seed_choice = gr.Radio(
                choices=["Fixed", "Random"],
                value="Random",
                label="Seed Type"
            )
            
            generate_btn = gr.Button(
                "🎨 Generate",
                variant="primary"
            )
            
        with gr.Column(scale=1):
            # Output
            output_image = gr.Image(
                label="Generated Image",
                type="filepath"
            )
    
    # События
    generate_btn.click(
        fn=generate_image,
        inputs=[prompt, width, height, steps, seed_choice],
        outputs=output_image
    )
    
    gr.Markdown(
        """
        ### Tips:
        - Try different prompts to get various results
        - Adjust width and height for different image sizes
        - Increase steps for better quality (but slower generation)
        - Use fixed seed to get reproducible results
        """
    )

# Запускаем интерфейс
demo.launch()