Spaces:

1-13-am
/

neural-style-transfer

Running

1-13-am commited on Oct 17, 2023

Commit

960296f

•

1 Parent(s): 4fa7ba9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from utils import transformer, tensor_to_img
 from network import Style_Transfer_Network
-check_point = torch.load("/content/check_point.pth", map_location = torch.device('cpu'))
 model = Style_Transfer_Network()
 model.load_state_dict(check_point['state_dict'])

 from utils import transformer, tensor_to_img
 from network import Style_Transfer_Network
+check_point = torch.load("check_point1_0.pth", map_location = torch.device('cpu'))
 model = Style_Transfer_Network()
 model.load_state_dict(check_point['state_dict'])