debug

app.py

@@ -15,8 +15,8 @@ def tensor_to_image(tensor):
     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.
+    content_image = content_image.astype(np.float32)[np.newaxis, ...] / 255.
+    style_image = style_image.astype(np.float32)[np.newaxis, ...] / 255.
     
     outputs = hub_module(tf.constant(content_image), tf.constant(style_image))
     stylized_image = outputs[0]