deviceをglobalに直す

q

@@ -0,0 +1,113 @@
+commit d7efa60788c2404153113982fe8b07cb980c2cd7 (HEAD -> main, origin/main)
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sun Jul 7 16:05:06 2024 +0900
+
+    fix
+
+commit b6c9f2e78b94c8ac2c962ff0fdde09f6a6fc093b (new_space/main)
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sun Jul 7 14:37:03 2024 +0900
+
+    Dockerfile(redis), model.py(dir), requirements.txt(redis,flask-limiter), app.py(bug fix)
+
+commit 90ca682e582d750492b3929edf3dcb55f74c739a
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sun Jul 7 12:15:55 2024 +0900
+
+    dos taisaku
+
+commit 64567a75ad58e5950612c36633efa6d7178c5fe0
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sun Jul 7 11:22:34 2024 +0900
+
+    Dockerfile to cpu
+
+commit bd4674a596783d10625ba1bd513a95855b9c7406
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sun Jul 7 10:58:31 2024 +0900
+
+    use_local編集
+
+commit d9b35ec1346875135a831cc142e68a49462c51a3
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sun Jul 7 00:22:41 2024 +0900
+
+    Dockerfile
+
+commit 4afd7b9e13992bbdc9f4f15edff10e246ce9f6dd
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 23:41:59 2024 +0900
+
+    Dockerfile
+
+commit 2a78754181e890bc05d5703b24251696e24c0e80
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 23:32:22 2024 +0900
+
+    Dockerfile
+
+commit 5f87f80cf45ac1f9ead6a3366442bf4ff94df64e
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 23:23:36 2024 +0900
+
+    Dockerfile
+
+commit d705298c38843dfffc3201fea345e63f1c46a4b5
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 23:11:56 2024 +0900
+
+    dotenv
+
+commit da165f522bb4e01e88871e693fe206ba5e555e45
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 23:03:57 2024 +0900
+
+    renamed:    Dockerfile.backend -> Dockerfile
+
+commit 7ef1daa3df22974a63a1a1bc4a728cd52fce840c (new_space/master, master)
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 22:55:32 2024 +0900
+
+    README.md
+
+commit 42a98e30be310706a986ab1aad437ff39e53335a
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 22:44:59 2024 +0900
+
+    README.md
+
+commit 3e40bc41313683f81c51df846b6b8446610023bc
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 22:41:39 2024 +0900
+
+    delete .env from commit
+
+commit 86c0504a689788ef1e74c890f700546c755b600c (origin/master, origin/HEAD)
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 22:05:16 2024 +0900
+
+    modify .gitignore to ignore images/
+
+commit 721391fed68569e481bf748ab3a4f41add5026cd
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 21:49:18 2024 +0900
+
+    prepare spaces
+
+commit dd5dfc863864f9a9d3e1f1b0667293a0932935cf
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 21:15:34 2024 +0900
+
+    docker fix
+
+commit d7a562ed99a39f214b899ca4f1ce5a9cb67c3d3b
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 14:41:18 2024 +0900
+
+    arg flag追加
+
+commit c9cc4410190cd2cf27dd50363e735b14c70323ab
+Author: Yeq6X <orange.m0912@gmail.com>
+Date:   Sat Jul 6 14:10:09 2024 +0900
+
+    init

scripts/anime.py

@@ -15,9 +15,10 @@ from PIL import Image
 import numpy as np
 
 model = None
+device = None
 
 def init_model(use_local=False):
-    global model
+    global model, device
     model_opt = "default"
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # issue: nevetherless, use_gpu is False, it still uses GPU
     model = create_model(model_opt, use_local).to(device)

scripts/model.py

@@ -155,7 +155,6 @@ def create_model(model, use_local):
     import os
     if model == 'default':
         model_path = (lambda filename, subfolder: os.path.join(subfolder, filename) if use_local else download_file(filename, subfolder))("netG.pth", "models/Anime2Sketch")
-        # model_path = ((filename, subfolder) => if (use_local) os.path.join(subfolder, filename) else download_file(filename, subfolder))("netG.pth", "models/Anime2Sketch") // JavaScript
 
         ckpt = torch.load(model_path)
         for key in list(ckpt.keys()):