Franco Astegiano commited on
Commit
9fa990b
·
unverified ·
1 Parent(s): d64bba4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -168,7 +168,7 @@ is the same as the content image shape.
168
  def modify(imageinput,style_input):
169
  #content_image = load_image(imageinput, content_img_size)
170
  #style_image = load_image(style_input, style_img_size)
171
- style_image = tf.nn.avg_pool(style_image, ksize=[3,3], strides=[1,1], padding='SAME')
172
  #show_n([content_image, style_image], ['Content image', 'Style image'])
173
  outputs = hub_module(tf.constant(imageinput), tf.constant(style_input))
174
  return outputs[0]
 
168
  def modify(imageinput,style_input):
169
  #content_image = load_image(imageinput, content_img_size)
170
  #style_image = load_image(style_input, style_img_size)
171
+ style_image = tf.nn.avg_pool(style_input, ksize=[3,3], strides=[1,1], padding='SAME')
172
  #show_n([content_image, style_image], ['Content image', 'Style image'])
173
  outputs = hub_module(tf.constant(imageinput), tf.constant(style_input))
174
  return outputs[0]