Update app.py

app.py

@@ -58,6 +58,17 @@ hidden_negative = "bad anatomy, disfigured, poorly drawn,deformed, mutation, mal
 hidden_cn_booster_text = ",漂亮的脸"
 hidden_cn_negative = ""
 
+hed = HEDdetector.from_pretrained('lllyasviel/ControlNet')
+
+controlnet_scribble = ControlNetModel.from_pretrained(
+    "lllyasviel/sd-controlnet-scribble", torch_dtype=torch.float16, safety_checker=None, requires_safety_checker=False, )
+depth_estimator = pipeline('depth-estimation')
+
+controlnet_depth = ControlNetModel.from_pretrained(
+    "lllyasviel/sd-controlnet-depth", torch_dtype=torch.float16
+)
+
+
 def translate(prompt):
     trans_text = prompt
     translated = model.generate(**tokenizer(trans_text, return_tensors="pt", padding=True))
@@ -70,10 +81,6 @@ def translate(prompt):
 def load_pipe_scribble():
     global pipe_scribble
     if pipe_scribble is None:
-        hed = HEDdetector.from_pretrained('lllyasviel/ControlNet')
-
-        controlnet_scribble = ControlNetModel.from_pretrained(
-            "lllyasviel/sd-controlnet-scribble", torch_dtype=torch.float16, safety_checker=None, requires_safety_checker=False, )
         
         pipe_scribble = StableDiffusionControlNetPipeline.from_single_file(
             "https://huggingface.co/shellypeng/anime-god/blob/main/animeGod_v10.safetensors", controlnet=controlnet_scribble, safety_checker=None, requires_safety_checker=False,
@@ -97,11 +104,6 @@ def load_pipe_scribble():
 def load_pipe_depth():
     global pipe_depth
     if pipe_depth is None:
-        depth_estimator = pipeline('depth-estimation')
-
-        controlnet_depth = ControlNetModel.from_pretrained(
-            "lllyasviel/sd-controlnet-depth", torch_dtype=torch.float16
-        )
         
         
         pipe_depth = StableDiffusionControlNetPipeline.from_single_file(