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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -95,9 +95,9 @@ def run_style_transfer(cnn, normalization_mean, normalization_std,
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])):
101
 
102
  #Start the timer
103
  start_time = time.time()
 
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.open(example_list[0]),styl_img=Image.open(example_list[1])):
101
 
102
  #Start the timer
103
  start_time = time.time()