.

app.py

@@ -10,6 +10,12 @@ 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
 
+def initialize_models():
+    # 初期化
+    initialize(_use_local=False, use_gpu=True, use_dotenv=True)
+    init_model(use_local=False)
+    load_wd14_tagger_model()
+
 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}")
@@ -205,11 +211,6 @@ with gr.Blocks() as demo:
     )
 
     # JavaScriptの読み込み
-    demo.load(None, None, None, js=script)
-
-# 初期化
-initialize(_use_local=False, use_gpu=True, use_dotenv=True)
-init_model(use_local=False)
-load_wd14_tagger_model()
+    demo.load(fn=initialize_models, js=script)
 
 demo.launch()