luca-martial commited on
Commit
da45ac7
1 Parent(s): fecdd34
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,10 +11,10 @@ def stylize(content_image, style_image):
11
  content_image = content_image.astype(np.float32) / 255.
12
  style_image = style_image.astype(np.float32) / 255.
13
 
14
- outputs = hub_module(tf.constant(content_image), tf.constant(style_image))
15
- stylized_image = outputs[0]
16
 
17
- return content_image.shape
18
 
19
 
20
  iface = gr.Interface(fn=stylize, inputs=["image", "image"], outputs="text")
 
11
  content_image = content_image.astype(np.float32) / 255.
12
  style_image = style_image.astype(np.float32) / 255.
13
 
14
+ #outputs = hub_module(tf.constant(content_image), tf.constant(style_image))
15
+ #stylized_image = outputs[0]
16
 
17
+ return str(content_image.shape)
18
 
19
 
20
  iface = gr.Interface(fn=stylize, inputs=["image", "image"], outputs="text")