Rooni commited on
Commit
1e89a47
·
verified ·
1 Parent(s): 3975802

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
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, 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 image_bytes, params_text, seed
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()