Update app.py

app.py

@@ -207,7 +207,8 @@ def load_model():
             model_path = f"/weights/{category_folder}/{character_name}/{model_name}"
             
             # 統合: 下記にしかない pickle_module の指定を追加
-            cpt = torch.load(f"/weights/{category_folder}/{character_name}/{model_name}", map_location="cpu", pickle_module=pickle)
+            cpt = torch.load(f"/weights/{category_folder}/{character_name}/{model_name}", map_location="cpu")
+            #cpt = torch.load(f"/weights/{category_folder}/{character_name}/{model_name}", map_location="cpu", pickle_module=pickle)
             
             model_version = cpt.get("version", "v1")
             print(f"Indexed model {model_title} by {model_author} ({model_version})")