Update app.py

app.py

@@ -107,9 +107,15 @@ def generate(prompt, negative_prompt, width=1024, height=1024, num_inference_ste
         image = (image * 255).astype(np.uint8)
         return Image.fromarray(image)
     finally:
-        if lora_id and lora_id.strip() != "":
-            pass
+        # if lora_id and lora_id.strip() != "":
+        #     pass
             # pipe.unload_lora_weights()
+        if clean_lora_id:
+            print(f"Unloading '{CUSTOM_LORA_NAME}' from this run.")
+            pipe.unload_lora_weights(CUSTOM_LORA_NAME)
+
+        # Always disable all active LoRAs to reset the state.
+        pipe.disable_lora()
 
 iface = gr.Interface(
     fn=generate,