Update app.py

app.py

@@ -101,7 +101,7 @@ 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),styl_img=Image.open(img_2)):
+def style_transfer(cont_img=Image.open(img_1).convert('RGB'),styl_img=Image.open(img_2).convert('RGB')):
   
   #Start the timer
   start_time = time.time()