fix

visualizer_drag_gradio.py

@@ -35,7 +35,7 @@ cache_dir = args.cache_dir
 
 device = 'cuda'
 IS_SPACE = "DragGan/DragGan" in os.environ.get('SPACE_ID', '')
-TIMEOUT = 60 if IS_SPACE else 0
+TIMEOUT = 80
 
 
 def reverse_point_pairs(points):
@@ -559,7 +559,7 @@ with gr.Blocks() as app:
             while True:
                 # add a TIMEOUT break
                 print(f'Running time: {time.time() - last_time}')
-                if time.time() - last_time > TIMEOUT:
+                if IS_SPACE and time.time() - last_time > TIMEOUT:
                     print('Timeout break!')
                     break
                 if global_state["temporal_params"]["stop"]: