Spaces:
Running
Running
Update examples/pixelArt/white_box_cartoonizer/cartoonize.py
Browse files
examples/pixelArt/white_box_cartoonizer/cartoonize.py
CHANGED
@@ -17,10 +17,10 @@ try:
|
|
17 |
except ImportError:
|
18 |
import tensorflow as tf
|
19 |
|
20 |
-
from
|
21 |
-
from
|
22 |
|
23 |
-
weights_dir = f'{os.getcwd()}/
|
24 |
gpu = len(sys.argv) < 2 or sys.argv[1] != '--cpu'
|
25 |
|
26 |
class WB_Cartoonize:
|
|
|
17 |
except ImportError:
|
18 |
import tensorflow as tf
|
19 |
|
20 |
+
from examples.pixelArt.white_box_cartoonizer.components.guided_filter import gf
|
21 |
+
from examples.pixelArt.white_box_cartoonizer.components.network import nk
|
22 |
|
23 |
+
weights_dir = f'{os.getcwd()}/examples/pixelArt/white_box_cartoonizer/saved_models'
|
24 |
gpu = len(sys.argv) < 2 or sys.argv[1] != '--cpu'
|
25 |
|
26 |
class WB_Cartoonize:
|