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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -94,8 +94,10 @@ def run_style_transfer(cnn, normalization_mean, normalization_std,
94
  return output_img
95
 
96
  #example_images
97
- example_list = [["examples/" + example] for example in os.listdir("examples")]
98
 
 
 
99
 
100
  #Defining the predict function
101
  def style_transfer(cont_img:Image.Image=None,styl_img:Image.Image=None) -> Image.Image:
 
94
  return output_img
95
 
96
  #example_images
97
+ example_list = [["examples/" + example] for example in os.listdir("examples")]
98
 
99
+ #Default output image
100
+ default_output_image = Image.fromarray(np.random.randint(0, 255, (image_height, image_width, 3), dtype=np.uint8))
101
 
102
  #Defining the predict function
103
  def style_transfer(cont_img:Image.Image=None,styl_img:Image.Image=None) -> Image.Image: