Rooni commited on
Commit
78c61d3
·
verified ·
1 Parent(s): 47c214a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -8
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, params_textbox):
32
  # Проверка очереди
33
  if len(queue) >= 150:
34
- return "Очередь переполнена. Пожалуйста, подождите.", None, None
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.update(images)
69
 
70
  # Удаляем задачу из очереди
71
  queue.remove(task_id)
72
 
73
- return images, params_text, seed
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: