Legola commited on
Commit
984fd8d
·
1 Parent(s): 1142e4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -95,9 +95,10 @@ 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.open(example_list[0]),styl_img=Image.open(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
+ img_1 = example_list[0]
99
+ img_2 = example_list[1]
100
  #Defining the predict function
101
+ def style_transfer(cont_img=Image.open(img_1),styl_img=Image.open(img_2)):
102
 
103
  #Start the timer
104
  start_time = time.time()