Update app.py

app.py

@@ -142,11 +142,11 @@ def get_imgs(base_image_path, style_reference_image_path):
   
   # Dimensions of the generated picture.
   width, height = base_image.size
-  img_nrows = 400
-  img_ncols = int(width * img_nrows / height)
   global img_nrows
   global img_ncols
-
+  img_nrows = 400
+  img_ncols = int(width * img_nrows / height)
+  
   iterations = 400
   for i in range(1, iterations + 1):
       loss, grads = compute_loss_and_grads(combination_image, base_image, style_reference_image)
@@ -154,7 +154,6 @@ def get_imgs(base_image_path, style_reference_image_path):
       if i % 100 == 0:
           print("Iteration %d: loss=%.2f" % (i, loss))
           img = deprocess_image(combination_image.numpy())
-
   return img