Freak-ppa commited on
Commit
56fef9c
·
verified ·
1 Parent(s): 4203cdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -84,7 +84,7 @@ def check_server_ready(port):
84
 
85
 
86
 
87
- @spaces.GPU(duration=240)
88
  def generate_image(prompt, image):
89
  prefix_filename = str(random.randint(0, 999999))
90
  prompt = prompt.replace('ComfyUI', prefix_filename)
@@ -112,7 +112,7 @@ def generate_image(prompt, image):
112
  start_queue(prompt, new_port)
113
 
114
  # Ожидание нового изображения
115
- timeout = 180 # Максимальное время ожидания в секундах
116
  start_time = time.time()
117
  while time.time() - start_time < timeout:
118
  latest_image = wait_for_image_with_prefix(OUTPUT_DIR, prefix_filename)
 
84
 
85
 
86
 
87
+ @spaces.GPU(duration=180)
88
  def generate_image(prompt, image):
89
  prefix_filename = str(random.randint(0, 999999))
90
  prompt = prompt.replace('ComfyUI', prefix_filename)
 
112
  start_queue(prompt, new_port)
113
 
114
  # Ожидание нового изображения
115
+ timeout = 240 # Максимальное время ожидания в секундах
116
  start_time = time.time()
117
  while time.time() - start_time < timeout:
118
  latest_image = wait_for_image_with_prefix(OUTPUT_DIR, prefix_filename)