Upload 2 files

app.py

@@ -701,7 +701,7 @@ def sd_gen_generate_pipeline(*args):
         None if item == "None" or item == "" else item # MOD
         for item in [args[7], args[9], args[11], args[13], args[15], args[17], args[19]]
     ]
-    lora_status = [None] * sd_gen.model.num_loras
+    lora_status = [None] * sd_gen.model.num_loras if getattr(sd_gen, "model", None) is not None else 7 # MOD
 
     msg_load_lora = "Updating LoRAs in GPU..."
     if load_lora_cpu:
@@ -1479,7 +1479,7 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
 
                     def change_visibility_canvas():
                         return gr.update(visible=True, interactive=True), gr.update(visible=False)
-                    show_canvas.click(change_visibility_canvas, [], [image_base, show_canvas])
+                    show_canvas.click(change_visibility_canvas, [], [image_base, show_canvas], show_api=False)
 
                     invert_mask = gr.Checkbox(value=False, label="Invert mask")
                     btn = gr.Button("Create mask")
@@ -1493,7 +1493,7 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
 
                 def send_img(img_source, img_result):
                     return img_source, img_result
-                btn_send.click(send_img, [img_source, img_result], [image_control, image_mask_gui])
+                btn_send.click(send_img, [img_source, img_result], [image_control, image_mask_gui], show_api=False)
 
         with gr.Tab("PNG Info"):
             with gr.Row():

constants.py

@@ -7,12 +7,13 @@ from stablepy import (
     ALL_BUILTIN_UPSCALERS,
     IP_ADAPTERS_SD,
     IP_ADAPTERS_SDXL,
+    PROMPT_WEIGHT_OPTIONS_PRIORITY,
 )
 
 IS_ZERO_GPU = bool(os.getenv("SPACES_ZERO_GPU"))
 
 # - **Download Models**
-DOWNLOAD_MODEL = "https://huggingface.co/TechnoByte/MilkyWonderland/resolve/main/milkyWonderland_v40.safetensors"
+DOWNLOAD_MODEL = "https://huggingface.co/zuv0/test/resolve/main/milkyWonderland_v40.safetensors"
 
 # - **Download VAEs**
 DOWNLOAD_VAE = "https://huggingface.co/fp16-guy/anything_kl-f8-anime2_vae-ft-mse-840000-ema-pruned_blessed_clearvae_fp16_cleaned/resolve/main/vae-ft-mse-840000-ema-pruned_fp16.safetensors?download=true"
@@ -321,15 +322,9 @@ DIFFUSERS_CONTROLNET_MODEL = [
     # "InstantX/FLUX.1-dev-Controlnet-Canny",
 ]
 
-PROMPT_W_OPTIONS = [
-    ("Compel format: (word)weight", "Compel"),
-    ("Classic format: (word:weight)", "Classic"),
-    ("Classic-original format: (word:weight)", "Classic-original"),
-    ("Classic-no_norm format: (word:weight)", "Classic-no_norm"),
-    ("Classic-sd_embed format: (word:weight)", "Classic-sd_embed"),
-    ("Classic-ignore", "Classic-ignore"),
-    ("None", "None"),
-]
+PROMPT_W_OPTIONS = [(pwf, pwf) for pwf in PROMPT_WEIGHT_OPTIONS_PRIORITY]
+PROMPT_W_OPTIONS[0] = ("Classic format: (word:weight)", "Classic")
+PROMPT_W_OPTIONS[1] = ("Compel format: (word)weight", "Compel")
 
 WARNING_MSG_VAE = (
     "Use the right VAE for your model to maintain image quality. The wrong"