limit models, concurrency

visualizer_drag_gradio.py

@@ -159,21 +159,20 @@ def preprocess_mask_info(global_state, image):
 
 
 # filter large models running on SPACES
+allowed_checkpoints = []  # all checkpoints
 if IS_SPACE:
-    blocked_checkpoints = ["stylegan_human_v2_512.pkl",
+    allowed_checkpoints = ["stylegan_human_v2_512.pkl",
                            "stylegan2_dogs_1024_pytorch.pkl"]
-else:
-    blocked_checkpoints = []
 
 valid_checkpoints_dict = {
     f.name.split('.')[0]: str(f)
     for f in Path(cache_dir).glob('*.pkl')
-    if f.name not in blocked_checkpoints
+    if f.name in allowed_checkpoints or not IS_SPACE
 }
 print('Valid checkpoint file:')
 print(valid_checkpoints_dict)
 
-init_pkl = 'stylegan2_lions_512_pytorch'
+init_pkl = 'stylegan_human_v2_512'
 
 with gr.Blocks() as app:
     gr.Markdown("""
@@ -908,5 +907,5 @@ with gr.Blocks() as app:
 
 print("SHAReD: Start app", parser.parse_args())
 gr.close_all()
-app.queue(concurrency_count=5, max_size=200, api_open=False)
+app.queue(concurrency_count=2, max_size=200, api_open=False)
 app.launch(share=args.share, show_api=False)