.

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 import os
 import io
 from PIL import Image
@@ -12,6 +13,7 @@ from scripts.survey import handle_form_submission, handle_visit_choice, handle_p
 # 初期化
 initialize(_use_local=False, use_gpu=True, use_dotenv=True)
 
+@spaces.GPU
 def process_image(input_image, mode, weight1=None, weight2=None):
     tokyo_time = datetime.now(timezone('Asia/Tokyo')).strftime("%Y-%m-%d %H:%M:%S")  # 日本時間のタイムスタンプ
     print(f"[{tokyo_time}] Processing image with mode={mode}, weight1={weight1}, weight2={weight2}")

scripts/process_utils.py

@@ -39,8 +39,10 @@ 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 = "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
     use_local = _use_local
 
     print(f"\nDevice: {device}, Local model: {_use_local}\n")
@@ -205,7 +207,7 @@ def create_rgba_image(binary_image: np.ndarray, color: list) -> Image.Image:
     rgba_image[:, :, 3] = binary_image
     return Image.fromarray(rgba_image, 'RGBA')
 
-@spaces.GPU
+# @spaces.GPU
 def generate_sotai_image(input_image: Image.Image, output_width: int, output_height: int) -> Image.Image:
     input_image = ensure_rgb(input_image)
     global sotai_gen_pipe
@@ -224,7 +226,6 @@ def generate_sotai_image(input_image: Image.Image, output_width: int, output_hei
         # EasyNegativeV2の内容
         easy_negative_v2 = "(worst quality, low quality, normal quality:1.4), lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry, artist name, (bad_prompt_version2:0.8)"
 
-        print(f"{sotai_gen_pipe.device}で画像生成中...")
         output = sotai_gen_pipe(
             prompt,
             image=[input_image, input_image],
@@ -251,7 +252,7 @@ def generate_sotai_image(input_image: Image.Image, output_width: int, output_hei
             torch.cuda.empty_cache()
         gc.collect()
 
-@spaces.GPU
+# @spaces.GPU
 def generate_refined_image(prompt: str, original_image: Image.Image, output_width: int, output_height: int, weight1: float, weight2: float) -> Image.Image:
     original_image = ensure_rgb(original_image)
     global refine_gen_pipe