File size: 252 Bytes
a1ded73
 
 
 
 
 
 
1
2
3
4
5
6
7
import os
import tensorflow as tf
import tensorflow_hub as hub
os.environ['TFHUB_MODEL_LOAD_FORMAT'] = 'COMPRESSED'

hub_model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
tf.saved_model.save(hub_model, 'style/1')