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)
|