Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,6 @@ import numpy as np
|
|
3 |
import PIL.Image as Image
|
4 |
import random
|
5 |
import time
|
6 |
-
import threading
|
7 |
|
8 |
# Список моделей с отображаемыми названиями и внутренними именами
|
9 |
models = {
|
@@ -28,10 +27,10 @@ models = {
|
|
28 |
# Очередь на генерации
|
29 |
queue = []
|
30 |
|
31 |
-
def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed, gallery,
|
32 |
# Проверка очереди
|
33 |
if len(queue) >= 150:
|
34 |
-
return "Очередь переполнена. Пожалуйста, подождите.",
|
35 |
|
36 |
# Добавляем задачу в очередь
|
37 |
task_id = random.randint(100000, 999999)
|
@@ -51,9 +50,6 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
|
|
51 |
f"Гуидансе скейл: {guidance_scale}"
|
52 |
)
|
53 |
|
54 |
-
# Обновление параметров
|
55 |
-
params_textbox.update(params_text)
|
56 |
-
|
57 |
images = []
|
58 |
for step in range(steps):
|
59 |
# Имитация генерации изображения
|
@@ -65,12 +61,12 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
|
|
65 |
|
66 |
# Добавление изображения в галерею
|
67 |
images.append(image)
|
68 |
-
gallery
|
69 |
|
70 |
# Удаляем задачу из очереди
|
71 |
queue.remove(task_id)
|
72 |
|
73 |
-
return
|
74 |
|
75 |
# Создание интерфейса
|
76 |
with gr.Blocks() as demo:
|
|
|
3 |
import PIL.Image as Image
|
4 |
import random
|
5 |
import time
|
|
|
6 |
|
7 |
# Список моделей с отображаемыми названиями и внутренними именами
|
8 |
models = {
|
|
|
27 |
# Очередь на генерации
|
28 |
queue = []
|
29 |
|
30 |
+
def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed, gallery, params_text):
|
31 |
# Проверка очереди
|
32 |
if len(queue) >= 150:
|
33 |
+
return "Очередь переполнена. Пожалуйста, подождите.", gallery, params_text, seed
|
34 |
|
35 |
# Добавляем задачу в очередь
|
36 |
task_id = random.randint(100000, 999999)
|
|
|
50 |
f"Гуидансе скейл: {guidance_scale}"
|
51 |
)
|
52 |
|
|
|
|
|
|
|
53 |
images = []
|
54 |
for step in range(steps):
|
55 |
# Имитация генерации изображения
|
|
|
61 |
|
62 |
# Добавление изображения в галерею
|
63 |
images.append(image)
|
64 |
+
gallery = images
|
65 |
|
66 |
# Удаляем задачу из очереди
|
67 |
queue.remove(task_id)
|
68 |
|
69 |
+
return gallery, params_text, seed
|
70 |
|
71 |
# Создание интерфейса
|
72 |
with gr.Blocks() as demo:
|