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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,8 +58,8 @@ imshow(style_image, 'Style Image')
58
 
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)
 
58
 
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)