init

app.py

@@ -0,0 +1,12 @@
+import gradio as gr
+import tensorflow as tf
+import tensorflow_hub as hub
+
+hub_model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
+
+def stylize(content_image, style_image):
+    stylized_image = hub_model(tf.constant(content_image), tf.constant(style_image))[0]
+    return tensor_to_image(stylized_image)
+
+iface = gr.Interface(fn=stylize, inputs=["image", "image"], outputs="image")
+iface.launch()