Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -27,10 +27,10 @@ models = {
|
|
27 |
# Очередь на генерации
|
28 |
queue = []
|
29 |
|
30 |
-
def generate_image(model_name, width, height, seed, steps, guidance_scale, random_seed):
|
31 |
# Проверка очереди
|
32 |
if len(queue) >= 150:
|
33 |
-
return "Очередь переполнена. Пожалуйста, подождите.", None, None
|
34 |
|
35 |
# Добавляем задачу в очередь
|
36 |
task_id = random.randint(100000, 999999)
|
@@ -56,7 +56,14 @@ def generate_image(model_name, width, height, seed, steps, guidance_scale, rando
|
|
56 |
image_bytes = buffered.getvalue()
|
57 |
|
58 |
# Формирование текста параметров
|
59 |
-
params_text =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
60 |
|
61 |
return image_bytes, params_text, seed
|
62 |
|
@@ -64,6 +71,9 @@ def generate_image(model_name, width, height, seed, steps, guidance_scale, rando
|
|
64 |
with gr.Blocks() as demo:
|
65 |
gr.Markdown("# Генератор изображений")
|
66 |
|
|
|
|
|
|
|
67 |
with gr.Row():
|
68 |
model_dropdown = gr.Dropdown(list(models.keys()), label="Модель", value="FLUX.1-dev")
|
69 |
|
@@ -79,7 +89,7 @@ with gr.Blocks() as demo:
|
|
79 |
|
80 |
with gr.Row():
|
81 |
steps_slider = gr.Slider(1, 100, value=50, step=1, label="Шаги")
|
82 |
-
guidance_scale_slider = gr.Slider(0.1, 20.0, value=7.5, step=0.1, label="Гуидансе
|
83 |
|
84 |
generate_button = gr.Button("Сгенерировать изображение")
|
85 |
output_image = gr.Image(label="Сгенерированное изображение")
|
@@ -87,7 +97,16 @@ with gr.Blocks() as demo:
|
|
87 |
|
88 |
generate_button.click(
|
89 |
fn=generate_image,
|
90 |
-
inputs=[
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
91 |
outputs=[output_image, params_textbox, seed_slider]
|
92 |
)
|
93 |
|
@@ -107,4 +126,4 @@ demo = demo.add_css(hide_footer)
|
|
107 |
# Установка таймаута
|
108 |
demo.queue(api_name="/generate").set_config(max_size=150, api_open=False, max_threads=1, timeout=125)
|
109 |
|
110 |
-
demo.launch()
|
|
|
27 |
# Очередь на генерации
|
28 |
queue = []
|
29 |
|
30 |
+
def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed):
|
31 |
# Проверка очереди
|
32 |
if len(queue) >= 150:
|
33 |
+
return "Очередь переполнена. Пожалуйста, подождите.", None, None, None
|
34 |
|
35 |
# Добавляем задачу в очередь
|
36 |
task_id = random.randint(100000, 999999)
|
|
|
56 |
image_bytes = buffered.getvalue()
|
57 |
|
58 |
# Формирование текста параметров
|
59 |
+
params_text = (
|
60 |
+
f"Промпт: {prompt}\n"
|
61 |
+
f"Модель: {model_name}\n"
|
62 |
+
f"Размер: {width}x{height}\n"
|
63 |
+
f"Сид: {seed}\n"
|
64 |
+
f"Шаги: {steps}\n"
|
65 |
+
f"Гуидансе скейл: {guidance_scale}"
|
66 |
+
)
|
67 |
|
68 |
return image_bytes, params_text, seed
|
69 |
|
|
|
71 |
with gr.Blocks() as demo:
|
72 |
gr.Markdown("# Генератор изображений")
|
73 |
|
74 |
+
with gr.Row():
|
75 |
+
prompt_textbox = gr.Textbox(label="Промпт", placeholder="Введите ваш промпт здесь", lines=2)
|
76 |
+
|
77 |
with gr.Row():
|
78 |
model_dropdown = gr.Dropdown(list(models.keys()), label="Модель", value="FLUX.1-dev")
|
79 |
|
|
|
89 |
|
90 |
with gr.Row():
|
91 |
steps_slider = gr.Slider(1, 100, value=50, step=1, label="Шаги")
|
92 |
+
guidance_scale_slider = gr.Slider(0.1, 20.0, value=7.5, step=0.1, label="Гуидансе скейл")
|
93 |
|
94 |
generate_button = gr.Button("Сгенерировать изображение")
|
95 |
output_image = gr.Image(label="Сгенерированное изображение")
|
|
|
97 |
|
98 |
generate_button.click(
|
99 |
fn=generate_image,
|
100 |
+
inputs=[
|
101 |
+
prompt_textbox,
|
102 |
+
model_dropdown,
|
103 |
+
width_slider,
|
104 |
+
height_slider,
|
105 |
+
seed_slider,
|
106 |
+
steps_slider,
|
107 |
+
guidance_scale_slider,
|
108 |
+
random_seed_checkbox
|
109 |
+
],
|
110 |
outputs=[output_image, params_textbox, seed_slider]
|
111 |
)
|
112 |
|
|
|
126 |
# Установка таймаута
|
127 |
demo.queue(api_name="/generate").set_config(max_size=150, api_open=False, max_threads=1, timeout=125)
|
128 |
|
129 |
+
demo.launch()
|