Update gradio_app.py

gradio_app.py

@@ -12,12 +12,12 @@ device = 'cuda'
 
 
 SIZE_TO_CLICK_SIZE = {
-    1024: 5,
+    512: 5,
     256: 2
 }
 
 CKPT_SIZE = {
-    'stylegan2-ffhq-config-f.pt': 1024,
+    'stylegan2-ffhq-config-f.pt': 512,
     'stylegan2-cat-config-f.pt': 256,
     'stylegan2-church-config-f.pt': 256,
     'stylegan2-horse-config-f.pt': 256,
@@ -221,7 +221,7 @@ def main():
             'history': []
         })
         points = gr.State({'target': [], 'handle': []})
-        size = gr.State(1024)
+        size = gr.State(512)
 
         with gr.Row():
             with gr.Column(scale=0.3):