Legola commited on
Commit
aec593a
·
1 Parent(s): 822a587

Update app.py

Browse files

Default image height and width

Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -92,7 +92,11 @@ def run_style_transfer(cnn, normalization_mean, normalization_std,
92
  output_img = Image.fromarray((output_np * 255).astype(np.uint8))
93
 
94
  return output_img
95
-
 
 
 
 
96
  #example_images
97
  example_list = [["examples/" + example] for example in os.listdir("examples")]
98
 
 
92
  output_img = Image.fromarray((output_np * 255).astype(np.uint8))
93
 
94
  return output_img
95
+
96
+ # Define the dimensions of the random output image
97
+ image_width = 128
98
+ image_height = 128
99
+
100
  #example_images
101
  example_list = [["examples/" + example] for example in os.listdir("examples")]
102