Update app.py

app.py

@@ -45,7 +45,7 @@ radio_style = gr.Radio(style_images, label="Choose Style")
 def perform_neural_transfer(content_image_input, style_image_input):
 
     content_image = content_image_input.astype(np.float32)[np.newaxis, ...] / 255.
-    content_image = tf.image.resize(content_image, (256, 256))
+    content_image = tf.image.resize(content_image, (600, 400))
     
     style_image_input = style_urls[style_image_input]
     style_image_input = plt.imread(style_image_input)
@@ -56,7 +56,7 @@ def perform_neural_transfer(content_image_input, style_image_input):
     hub_module = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
 
     outputs = hub_module(tf.constant(content_image), tf.constant(style_image))
-    stylized_image = outputs[0].resize(outputs[0], (256, 256))
+    stylized_image = outputs[0]
 
     return tensor_to_image(stylized_image)