YueLu315 commited on
Commit
26b1026
1 Parent(s): 6786858

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ imshow(style_image, 'Style Image')
59
  style_path
60
 
61
  #hub_model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
62
- hub_model = hub.load('https://hub.tensorflow.google.cn/google/magenta/arbitrary-image-stylization-v1-256/2')
63
 
64
  stylized_image = hub_model(tf.constant(content_image), tf.constant(style_image))[0]
65
  tensor_to_image(stylized_image)
 
59
  style_path
60
 
61
  #hub_model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
62
+ hub_model = hub.load('https://hub.tensorflow.google.cn/sayakpaul/lite-model/arbitrary-image-stylization-inceptionv3-dynamic-shapes/dr/predict/1')
63
 
64
  stylized_image = hub_model(tf.constant(content_image), tf.constant(style_image))[0]
65
  tensor_to_image(stylized_image)