Update app.py

app.py

@@ -203,7 +203,8 @@ 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")
+            import pickle
+            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))