Legola commited on
Commit
2de3b07
·
1 Parent(s): cfee608

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -92,7 +92,9 @@ 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
 
97
  #Defining the predict function
98
  def style_transfer(cont_img=Image(example_list[0]),styl_img=Image(example_list[1])):
 
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
 
99
  #Defining the predict function
100
  def style_transfer(cont_img=Image(example_list[0]),styl_img=Image(example_list[1])):