Update app.py
Browse files
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))
|
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()
|