debug

app.py

@@ -6,16 +6,23 @@ import numpy as np
 
 hub_model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2') 
 
+def tensor_to_image(tensor):
+    tensor = tensor*255
+    tensor = np.array(tensor, dtype=np.uint8)
+    if np.ndim(tensor)>3:
+      assert tensor.shape[0] == 1
+      tensor = tensor[0]
+    return PIL.Image.fromarray(tensor)
 
 def stylize(content_image, style_image):
     content_image = content_image.astype(np.float32) / 255.
     style_image = style_image.astype(np.float32) / 255.
     
-    #outputs = hub_module(tf.constant(content_image), tf.constant(style_image))
-    #stylized_image = outputs[0]
+    outputs = hub_module(tf.constant(content_image), tf.constant(style_image))
+    stylized_image = outputs[0]
 
-    return str(content_image.shape)
+    return tensor_to_image(stylized_image)
     
 
-iface = gr.Interface(fn=stylize, inputs=["image", "image"], outputs="text")
+iface = gr.Interface(fn=stylize, inputs=["image", "image"], outputs="image")
 iface.launch()