Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -84,7 +84,7 @@ def check_server_ready():
|
|
84 |
|
85 |
|
86 |
queue_reqs = set()
|
87 |
-
|
88 |
|
89 |
@spaces.GPU(duration=240)
|
90 |
def generate_image(prompt, image):
|
@@ -97,10 +97,13 @@ def generate_image(prompt, image):
|
|
97 |
|
98 |
queue_reqs.add(prefix_filename)
|
99 |
process = None
|
|
|
100 |
try:
|
101 |
-
|
102 |
-
|
103 |
-
|
|
|
|
|
104 |
|
105 |
# Ожидание запуска сервера
|
106 |
for _ in range(20): # Максимум 20 секунд ожидания
|
@@ -139,6 +142,7 @@ def generate_image(prompt, image):
|
|
139 |
process.wait(timeout=5)
|
140 |
except subprocess.TimeoutExpired:
|
141 |
process.kill()
|
|
|
142 |
|
143 |
|
144 |
if __name__ == "__main__":
|
|
|
84 |
|
85 |
|
86 |
queue_reqs = set()
|
87 |
+
process_started = False
|
88 |
|
89 |
@spaces.GPU(duration=240)
|
90 |
def generate_image(prompt, image):
|
|
|
97 |
|
98 |
queue_reqs.add(prefix_filename)
|
99 |
process = None
|
100 |
+
global process_started
|
101 |
try:
|
102 |
+
if not process_started:
|
103 |
+
process_started = True
|
104 |
+
# Запускаем скрипт как подпроцесс
|
105 |
+
process = subprocess.Popen([sys.executable, COMF_PATH, "--listen", "127.0.0.1"])
|
106 |
+
logger.debug(f'Subprocess started with PID: {process.pid}')
|
107 |
|
108 |
# Ожидание запуска сервера
|
109 |
for _ in range(20): # Максимум 20 секунд ожидания
|
|
|
142 |
process.wait(timeout=5)
|
143 |
except subprocess.TimeoutExpired:
|
144 |
process.kill()
|
145 |
+
process_started = False
|
146 |
|
147 |
|
148 |
if __name__ == "__main__":
|