Update app.py

app.py

@@ -139,6 +139,11 @@ def get_imgs(base_image_path, style_reference_image_path):
   base_image = preprocess_image(base_image_path)
   style_reference_image = preprocess_image(style_reference_image_path)
   combination_image = tf.Variable(preprocess_image(base_image_path))
+  
+  # Dimensions of the generated picture.
+  width, height = base_image.size
+  img_nrows = 400
+  img_ncols = int(width * img_nrows / height)
 
   iterations = 400
   for i in range(1, iterations + 1):