Rooni commited on
Commit
33df38a
·
verified ·
1 Parent(s): 1a41a9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -6
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 = f"Модель: {model_name}\nРазмер: {width}x{height}\nСид: {seed}\nШаги: {steps}\nГуидансе скала: {guidance_scale}"
 
 
 
 
 
 
 
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=[model_dropdown, width_slider, height_slider, seed_slider, steps_slider, guidance_scale_slider, random_seed_checkbox],
 
 
 
 
 
 
 
 
 
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()