MAINT: pretrained models

app.py

@@ -12,14 +12,12 @@ from torch.autograd import Variable
 from network.Transformer import Transformer
 
 LOAD_SIZE = 1280
-STYLE = "Shinkai"
-MODEL_PATH = "pretrained_model"
+STYLE = "shinkai_makoto"
+MODEL_PATH = "models"
 COLOUR_MODEL = "RGB"
 
 model = Transformer()
-model.load_state_dict(
-    torch.load(os.path.join(MODEL_PATH, f"{STYLE}_net_G_float.pth"))
-)
+model.load_state_dict(torch.load(os.path.join(MODEL_PATH, f"{STYLE}.pth")))
 model.eval()
 
 disable_gpu = torch.cuda.is_available()

models/hosoda_mamoru.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c666eea7700864d5972765cc43e926d900174648297bfef494006dc230fd1bf0
+size 44529096

models/kon_satoshi.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0629352a54838e56a2ad7fca3e6e51e6889d4338c37469f9ddb43e5929ef9475
+size 44529096

models/miyazaki_hayao.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ab0e492efb3b705487db38679e363dc8b1f016692913bbe100587d695a9e2b5
+size 44529096

models/shinkai_makoto.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c3547f611e780e79aebde7f7bc2b6c278555d701620f125583d666351044c486
+size 44529096