Update app.py

app.py

@@ -136,17 +136,17 @@ optimizer = keras.optimizers.SGD(
 )
 
 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
+  width, height = keras.preprocessing.image.load_img(base_image_path).size
   global img_nrows
   global img_ncols
   img_nrows = 400
   img_ncols = int(width * img_nrows / height)
   
+  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))
+  
   iterations = 400
   for i in range(1, iterations + 1):
       loss, grads = compute_loss_and_grads(combination_image, base_image, style_reference_image)