Spaces:
Paused
Paused
Update gradio_app.py
Browse files- gradio_app.py +3 -3
gradio_app.py
CHANGED
@@ -12,12 +12,12 @@ device = 'cuda'
|
|
12 |
|
13 |
|
14 |
SIZE_TO_CLICK_SIZE = {
|
15 |
-
|
16 |
256: 2
|
17 |
}
|
18 |
|
19 |
CKPT_SIZE = {
|
20 |
-
'stylegan2-ffhq-config-f.pt':
|
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(
|
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):
|