cthleen commited on
Commit
849c834
·
1 Parent(s): f1fb0c8

fix error

Browse files
Files changed (2) hide show
  1. app.py +1 -2
  2. stylegan2/legacy.py +1 -1
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import torch
2
  from stylegan2 import legacy
3
- from stylegan2.dnnlib import util
4
  import numpy as np
5
  from PIL import Image
6
  import gradio as gr
@@ -28,7 +27,7 @@ gr.Interface(
28
  fn=generate_image,
29
  inputs=[],
30
  outputs="image",
31
- title="StyleGAN2 Generator",
32
  description="Click to generate a random image",
33
  allow_flagging="never"
34
  ).launch()
 
1
  import torch
2
  from stylegan2 import legacy
 
3
  import numpy as np
4
  from PIL import Image
5
  import gradio as gr
 
27
  fn=generate_image,
28
  inputs=[],
29
  outputs="image",
30
+ title="StyleGAN2 Batik Generator",
31
  description="Click to generate a random image",
32
  allow_flagging="never"
33
  ).launch()
stylegan2/legacy.py CHANGED
@@ -12,7 +12,7 @@ import re
12
  import copy
13
  import numpy as np
14
  import torch
15
- import dnnlib
16
  from torch_utils import misc
17
 
18
  #----------------------------------------------------------------------------
 
12
  import copy
13
  import numpy as np
14
  import torch
15
+ from . import dnnlib
16
  from torch_utils import misc
17
 
18
  #----------------------------------------------------------------------------