Legola commited on
Commit
e87d8a8
·
1 Parent(s): b39f105

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.Image=torch.zeros((1,128,128)).detach().numpy(),styl_img:Image.Image=torch.zeros((1,128,128))).detach.numpy() -> Image.Image:
102
 
103
  #Start the timer
104
  start_time = time.time()
 
98
 
99
 
100
  #Defining the predict function
101
+ def style_transfer(cont_img:Image.Image=torch.zeros((1,128,128)).detach().numpy(),styl_img:Image.Image=torch.zeros((1,128,128)).detach().numpy()) -> Image.Image:
102
 
103
  #Start the timer
104
  start_time = time.time()