Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,6 @@ import gradio as gr
|
|
2 |
import numpy as np
|
3 |
import PIL.Image as Image
|
4 |
import random
|
5 |
-
import io
|
6 |
import time
|
7 |
|
8 |
# Список моделей с отображаемыми названиями и внутренними именами
|
@@ -31,7 +30,7 @@ queue = []
|
|
31 |
def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed):
|
32 |
# Проверка очереди
|
33 |
if len(queue) >= 150:
|
34 |
-
return "Очередь переполнена. Пожалуйста, подождите.", None, None
|
35 |
|
36 |
# Добавляем задачу в очередь
|
37 |
task_id = random.randint(100000, 999999)
|
@@ -51,11 +50,6 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
|
|
51 |
image_array = np.random.randint(0, 256, (height, width, 3), dtype=np.uint8)
|
52 |
image = Image.fromarray(image_array)
|
53 |
|
54 |
-
# Сохранение изображения в буфер
|
55 |
-
buffered = io.BytesIO()
|
56 |
-
image.save(buffered, format="PNG")
|
57 |
-
image_bytes = buffered.getvalue()
|
58 |
-
|
59 |
# Формирование текста параметров
|
60 |
params_text = (
|
61 |
f"Промпт: {prompt}\n"
|
@@ -66,7 +60,7 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
|
|
66 |
f"Гуидансе скейл: {guidance_scale}"
|
67 |
)
|
68 |
|
69 |
-
return
|
70 |
|
71 |
# Создание интерфейса
|
72 |
with gr.Blocks() as demo:
|
@@ -127,4 +121,4 @@ demo = demo.add_css(hide_footer)
|
|
127 |
# Установка таймаута
|
128 |
demo.queue(api_name="/generate").set_config(max_size=150, api_open=False, max_threads=1, timeout=125)
|
129 |
|
130 |
-
demo.launch()
|
|
|
2 |
import numpy as np
|
3 |
import PIL.Image as Image
|
4 |
import random
|
|
|
5 |
import time
|
6 |
|
7 |
# Список моделей с отображаемыми названиями и внутренними именами
|
|
|
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
|
34 |
|
35 |
# Добавляем задачу в очередь
|
36 |
task_id = random.randint(100000, 999999)
|
|
|
50 |
image_array = np.random.randint(0, 256, (height, width, 3), dtype=np.uint8)
|
51 |
image = Image.fromarray(image_array)
|
52 |
|
|
|
|
|
|
|
|
|
|
|
53 |
# Формирование текста параметров
|
54 |
params_text = (
|
55 |
f"Промпт: {prompt}\n"
|
|
|
60 |
f"Гуидансе скейл: {guidance_scale}"
|
61 |
)
|
62 |
|
63 |
+
return image, params_text, seed
|
64 |
|
65 |
# Создание интерфейса
|
66 |
with gr.Blocks() as demo:
|
|
|
121 |
# Установка таймаута
|
122 |
demo.queue(api_name="/generate").set_config(max_size=150, api_open=False, max_threads=1, timeout=125)
|
123 |
|
124 |
+
demo.launch()
|