Legola commited on
Commit
f176b2b
·
1 Parent(s): 0799656

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
98
 
99
 
100
  #Defining the predict function
101
- def style_transfer(cont_img=Image.open(example_list[0][0]).convert('RGB'),styl_img=Image.open(example_list[1][0]).convert('RGB')):
102
 
103
  #Start the timer
104
  start_time = time.time()
 
98
 
99
 
100
  #Defining the predict function
101
+ def style_transfer(cont_img:PIL.Image.Image,styl_img:PIL.Image.Image) -> PIL.Image.Image:
102
 
103
  #Start the timer
104
  start_time = time.time()