Spaces:
Building
Building
Update app.py
Browse files
app.py
CHANGED
@@ -84,7 +84,7 @@ def check_server_ready(port):
|
|
84 |
|
85 |
|
86 |
|
87 |
-
@spaces.GPU(duration=
|
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 =
|
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)
|