Update app.py

app.py

@@ -174,7 +174,7 @@ def infer(name, path, index, vc_input, vc_upload, tts_text, tts_voice, f0_up_key
         info = traceback.format_exc()
         print(info)
         return info, (None, None)
-
+import pickle
 def load_model():
     categories = []
     with open("/weights/folder_info.json", "r", encoding="utf-8") as f:
@@ -203,7 +203,7 @@ def load_model():
             if not (model_author in authors or "/" in model_author or "&" in model_author):
                 authors.append(model_author)
             model_path =  f"/weights/{category_folder}/{character_name}/{model_name}"
-            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})")
             models.append((character_name, model_title, model_author, model_cover, model_version, model_path, model_index))