Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -58,8 +58,8 @@ imshow(style_image, 'Style Image')
|
|
58 |
|
59 |
style_path
|
60 |
|
61 |
-
|
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)
|