Update app.py

app.py

@@ -91,31 +91,6 @@ def is_first_time_load():
         return True
     return False
 
-# スタートアップ時のGPUリソース確保
-if IN_HF_SPACE and 'spaces' in globals() and not cpu_fallback_mode:
-    # スタートアップ時にGPUリソースを確保するダミー関数
-    @spaces.GPU(duration=300)
-    def reserve_gpu_resource():
-        """スタートアップ時にGPUリソースを確保するダミー関数"""
-        print("GPU resources reserved for the application")
-        # 実際にはここで小さなGPU操作を行ってリソースを確保
-        if torch.cuda.is_available():
-            try:
-                test_tensor = torch.zeros(1, device='cuda')
-                test_tensor = test_tensor + 1
-                del test_tensor
-                torch.cuda.empty_cache()
-                return True
-            except Exception as e:
-                print(f"GPU予約中にエラー: {e}")
-                global cpu_fallback_mode
-                cpu_fallback_mode = True
-                return False
-        return False
-    
-    # アプリ起動前にGPUリソースを確保（最小限の操作のみ）
-    reserve_gpu_resource()
-
 
 # GPU制限を超えたかどうかを確認する関数
 def check_gpu_quota_exceeded():
@@ -1290,7 +1265,7 @@ def worker_with_temp_files(
 
 # 統合版プロセス関数 - GPU/CPU両方に対応（Spaces環境向け改良版）
 if IN_HF_SPACE and 'spaces' in globals():
-    @spaces.GPU(duration=300)
+    @spaces.GPU(duration=120)
     def process_with_temp(image_mask_dict, lora_multiplier=1.0):
         """一時ファイルを使用する処理メインフロー（GPU/CPU対応）"""
         global stream, cpu_fallback_mode
@@ -1590,8 +1565,32 @@ else:
                 cleanup_temp_files(temp_dir)
             raise e
 
+
+# 処理終了時に明示的にGPUメモリを解放
+def cleanup_gpu_resources():
+    """GPUリソースを明示的に解放する"""
+    global models
+    
+    # モデルを全てCPUに移動
+    for model_name, model in models.items():
+        try:
+            if model is not None and hasattr(model, 'to') and callable(model.to):
+                model.to('cpu')
+                print(f"{model_name}をCPUに移動しました")
+        except Exception as e:
+            print(f"{model_name}のCPU移動中にエラー: {e}")
+    
+    # キャッシュクリア
+    try:
+        torch.cuda.empty_cache()
+        gc.collect()
+        print("GPUキャッシュをクリアしました")
+    except Exception as e:
+        print(f"GPUキャッシュクリア中にエラー: {e}")
+
 # 処理中止・クリーンアップ関数（GPU/CPU共通）
 def end_process_with_cleanup():
+    cleanup_gpu_resources()
     """処理を中止し、一時ファイルを削除する"""
     global stream
     
@@ -1640,7 +1639,7 @@ with block:
 
         with gr.Column():
             preview_image = gr.Image(label="生成プレビュー", visible=False)
-            result_frame = gr.Image(label="生成結果", visible=False, type="pil",height="60vh")
+            result_frame = gr.Image(label="生成結果", visible=False, type="pil", height="60vh")
             
             progress_desc = gr.Markdown("", elem_classes="no-generating-animation")
             progress_bar = gr.HTML("", elem_classes="no-generating-animation")