fix
Browse files
visualizer_drag_gradio.py
CHANGED
@@ -35,7 +35,7 @@ cache_dir = args.cache_dir
|
|
35 |
|
36 |
device = 'cuda'
|
37 |
IS_SPACE = "DragGan/DragGan" in os.environ.get('SPACE_ID', '')
|
38 |
-
TIMEOUT =
|
39 |
|
40 |
|
41 |
def reverse_point_pairs(points):
|
@@ -559,7 +559,7 @@ with gr.Blocks() as app:
|
|
559 |
while True:
|
560 |
# add a TIMEOUT break
|
561 |
print(f'Running time: {time.time() - last_time}')
|
562 |
-
if time.time() - last_time > TIMEOUT:
|
563 |
print('Timeout break!')
|
564 |
break
|
565 |
if global_state["temporal_params"]["stop"]:
|
|
|
35 |
|
36 |
device = 'cuda'
|
37 |
IS_SPACE = "DragGan/DragGan" in os.environ.get('SPACE_ID', '')
|
38 |
+
TIMEOUT = 80
|
39 |
|
40 |
|
41 |
def reverse_point_pairs(points):
|
|
|
559 |
while True:
|
560 |
# add a TIMEOUT break
|
561 |
print(f'Running time: {time.time() - last_time}')
|
562 |
+
if IS_SPACE and time.time() - last_time > TIMEOUT:
|
563 |
print('Timeout break!')
|
564 |
break
|
565 |
if global_state["temporal_params"]["stop"]:
|