wip

app.py

@@ -164,7 +164,7 @@ def make_canny_condition(image, min_val=100, max_val=200, w_bilateral=True):
 default_negative_prompt = "Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"
 
 # Global variable to track the currently loaded LoRA
-current_lora_name = None
+CURRENT_LORA_NAME = None
 
 # Load face detection and recognition package
 app = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
@@ -316,12 +316,12 @@ def generate_image(image_path, prompt, num_steps, guidance_scale, seed, num_imag
     #     print("Using LoRA: ", lora_name)
 
 
-    if lora_name != current_lora_name:  # Check if LoRA needs to be changed
-        if current_lora_name is not None:  # If a LoRA is already loaded, unload it
+    if lora_name != CURRENT_LORA_NAME:  # Check if LoRA needs to be changed
+        if CURRENT_LORA_NAME is not None:  # If a LoRA is already loaded, unload it
             pipe.disable_lora()
             pipe.unfuse_lora()
             pipe.unload_lora_weights()
-            print(f"Unloaded LoRA: {current_lora_name}")
+            print(f"Unloaded LoRA: {CURRENT_LORA_NAME}")
 
         if lora_name != "":  # Load the new LoRA if specified
             lora_path = os.path.join(lora_base_path, lora_name, "pytorch_lora_weights.safetensors")
@@ -334,7 +334,7 @@ def generate_image(image_path, prompt, num_steps, guidance_scale, seed, num_imag
             print(f"Loaded new LoRA: {lora_name}")
 
         # Update the current LoRA name
-        current_lora_name = lora_name
+        CURRENT_LORA_NAME = lora_name
 
     if lora_name != "":
         full_prompt = f"{lora_prefix} {prompt}"