Spaces:

AfrodreamsAI
/

afrodreams

Runtime error

Owos commited on Sep 26, 2022

Commit

ca8ddf3

•

1 Parent(s): 70b4c14

changed to nin model

Files changed (1) hide show

neural_style.py CHANGED Viewed

@@ -77,7 +77,7 @@ class TransferParams():
     style_scale = 1.0
     original_colors = 0
     pooling = 'max'
-    model_file = 'models/vgg19-d01eb7cb.pth'
     disable_check = False
     backend = 'mkl'
     cudnn_autotune = False

     style_scale = 1.0
     original_colors = 0
     pooling = 'max'
+    model_file = 'models/nin_imagenet.pth'
     disable_check = False
     backend = 'mkl'
     cudnn_autotune = False