.

app.py

@@ -9,6 +9,9 @@ from scripts.anime import init_model
 from datetime import datetime
 from pytz import timezone
 from scripts.survey import handle_form_submission, handle_visit_choice, handle_proceed, localize, script, generate_image, send_feedback
+import torch
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # 初期化
 initialize(_use_local=False, use_gpu=True, use_dotenv=True)

scripts/anime.py

@@ -27,7 +27,7 @@ def init_model(use_local=False):
     model.eval()
 
 # numpy配列の画像を受け取り、線画を生成してnumpy配列で返す
-@spaces.GPU
+# @spaces.GPU
 def generate_sketch(image, clahe_clip=-1, load_size=512):
     """
     Generate sketch image from input image

scripts/process_utils.py

@@ -39,10 +39,8 @@ def initialize(_use_local=False, use_gpu=False, use_dotenv=False):
     if use_dotenv:
         load_dotenv()
     global model, sotai_gen_pipe, refine_gen_pipe, use_local, device, torch_dtype
-    # device = "cuda" if use_gpu and torch.cuda.is_available() else "cpu"
-    # torch_dtype = torch.float16 if device == "cuda" else torch.float32
-    device = torch.device('cuda')
-    torch_dtype = torch.float16
+    device = "cuda" if use_gpu and torch.cuda.is_available() else "cpu"
+    torch_dtype = torch.float16 if device == "cuda" else torch.float32
     use_local = _use_local
 
     print(f"\nDevice: {device}, Local model: {_use_local}\n")