Hugging Face's logo

Spaces:
AfrodreamsAI
/
afrodreams
Runtime error

App Files Community
Owos commited on
Commit
6fe0a06
·
1 Parent(s): 090fcd3

Update neural_style.py

Browse files
Files changed (1) hide show
  1. neural_style.py +3 -3
neural_style.py CHANGED
@@ -77,13 +77,13 @@ class TransferParams():
77
  style_scale = 1.0
78
  original_colors = 0
79
  pooling = 'max'
80
- model_file = 'models/nin_imagenet.pth'
81
  disable_check = False
82
  backend = 'mkl'
83
  cudnn_autotune = False
84
  seed = -1
85
- content_layers = 'relu0,relu3,relu7,relu12'
86
- style_layers = 'relu0,relu3,relu7,relu12'
87
  multidevice_strategy = '4,7,29'
88
 
89
  def main():
 
77
  style_scale = 1.0
78
  original_colors = 0
79
  pooling = 'max'
80
+ model_file = 'models/vgg16-00b39a1b.pth'#nin_imagenet.pth'
81
  disable_check = False
82
  backend = 'mkl'
83
  cudnn_autotune = False
84
  seed = -1
85
+ content_layers = 'relu4_2'#relu0,relu3,relu7,relu12'
86
+ style_layers = 'relu1_1,relu2_1,relu3_1,relu4_1,relu5_1'#relu0,relu3,relu7,relu12'
87
  multidevice_strategy = '4,7,29'
88
 
89
  def main():