Update app.py

app.py

@@ -95,13 +95,10 @@ def run_style_transfer(cnn, normalization_mean, normalization_std,
                            
 #example_images
 example_list = [["examples/" + example] for example in os.listdir("examples")]   
-print(example_list)
-img_1 = example_list[0][0]
-img_2 = example_list[1][0]
-print(img_1)
-print(img_2)
+
+
 #Defining the predict function
-def style_transfer(cont_img=Image.open(img_1).convert('RGB'),styl_img=Image.open(img_2).convert('RGB')):
+def style_transfer(cont_img=Image.open(example_list[0][0]).convert('RGB'),styl_img=Image.open(example_list[1][0]).convert('RGB')):
   
   #Start the timer
   start_time = time.time()