Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
-import swift.llm  # swift.llm パッケージを直接インポート
 
 # 使用可能なモデルのリスト
 models = ["Qwen/Qwen2.5-7B-Instruct", "Qwen/Qwen2.5-14B-Instruct", "Qwen/Qwen2.5-32B-Instruct"]
 
+# モデルを事前にロード
+model_pipelines = {model: pipeline(task=Tasks.text_generation, model=model) for model in models}
+
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -15,11 +17,8 @@ def respond(
     top_p,
     selected_model
 ):
-    # 型変換: selected_modelを文字列に変換
-    selected_model = str(selected_model)
-    
-    # 選択したモデルに基づいてPipelineを初期化
-    pipe = pipeline(task=Tasks.text_generation, model=selected_model)
+    # 選択したモデルに基づいてPipelineを選択
+    pipe = model_pipelines[selected_model]
 
     messages = [{"role": "system", "content": system_message}]