shainis commited on
Commit
b382940
1 Parent(s): 0daacc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ radio_style = gr.Radio(style_images, label="Choose Style")
45
  def perform_neural_transfer(content_image_input, style_image_input):
46
 
47
  content_image = content_image_input.astype(np.float32)[np.newaxis, ...] / 255.
48
- content_image = tf.image.resize(content_image, (600, 400))
49
 
50
  style_image_input = style_urls[style_image_input]
51
  style_image_input = plt.imread(style_image_input)
 
45
  def perform_neural_transfer(content_image_input, style_image_input):
46
 
47
  content_image = content_image_input.astype(np.float32)[np.newaxis, ...] / 255.
48
+ content_image = tf.image.resize(content_image, (400, 600))
49
 
50
  style_image_input = style_urls[style_image_input]
51
  style_image_input = plt.imread(style_image_input)