fffiloni commited on
Commit
c9bea79
1 Parent(s): 56d974d

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +3 -3
gradio_app.py CHANGED
@@ -12,12 +12,12 @@ device = 'cuda'
12
 
13
 
14
  SIZE_TO_CLICK_SIZE = {
15
- 1024: 5,
16
  256: 2
17
  }
18
 
19
  CKPT_SIZE = {
20
- 'stylegan2-ffhq-config-f.pt': 1024,
21
  'stylegan2-cat-config-f.pt': 256,
22
  'stylegan2-church-config-f.pt': 256,
23
  'stylegan2-horse-config-f.pt': 256,
@@ -221,7 +221,7 @@ def main():
221
  'history': []
222
  })
223
  points = gr.State({'target': [], 'handle': []})
224
- size = gr.State(1024)
225
 
226
  with gr.Row():
227
  with gr.Column(scale=0.3):
 
12
 
13
 
14
  SIZE_TO_CLICK_SIZE = {
15
+ 512: 5,
16
  256: 2
17
  }
18
 
19
  CKPT_SIZE = {
20
+ 'stylegan2-ffhq-config-f.pt': 512,
21
  'stylegan2-cat-config-f.pt': 256,
22
  'stylegan2-church-config-f.pt': 256,
23
  'stylegan2-horse-config-f.pt': 256,
 
221
  'history': []
222
  })
223
  points = gr.State({'target': [], 'handle': []})
224
+ size = gr.State(512)
225
 
226
  with gr.Row():
227
  with gr.Column(scale=0.3):