Hugging Face's logo

Spaces:
cthleen
/
batik-test2
Sleeping

App Files Community
cthleen commited on
Commit
fc626a1
·
1 Parent(s): c1c640d

add stylegan

Browse files
Files changed (1) hide show
  1. stylegan_generator.py +1 -3
stylegan_generator.py CHANGED
@@ -6,13 +6,11 @@ import onnxruntime as ort
6
  from PIL import Image
7
  import gradio as gr
8
 
9
- MODEL_NAME = 'batik_stylegan.onnx'
10
- CHANNELS = 3
11
  LATENT_FEATURES = 512
12
  RESOLUTION = 256
13
  LAST_INDEX = math.log2(RESOLUTION) - 2
14
 
15
- MODEL_PATH = os.path.join("model", "")
16
 
17
  alpha = np.array([1.0], dtype=np.float32)
18
  steps = np.array([LAST_INDEX], dtype=np.int64)
 
6
  from PIL import Image
7
  import gradio as gr
8
 
 
 
9
  LATENT_FEATURES = 512
10
  RESOLUTION = 256
11
  LAST_INDEX = math.log2(RESOLUTION) - 2
12
 
13
+ MODEL_PATH = os.path.join("model", "batik_stylegan.onnx")
14
 
15
  alpha = np.array([1.0], dtype=np.float32)
16
  steps = np.array([LAST_INDEX], dtype=np.int64)