Update app.py

app.py

@@ -33,6 +33,15 @@ def feifeimodload():
         hf_hub_download("aifeifei798/sldr_flux_nsfw_v2-studio", "sldr_flux_nsfw_v2-studio.safetensors"),
         adapter_name="sldr_flux_nsfw_v2",
     )
+
+    pipe.load_lora_weights(
+        hf_hub_download(
+            "aifeifei798/big-boobs-clothed",
+            "big-boobs-clothed-v2.safetensors",
+        ),
+        adapter_name="big-boobs-clothed-v2",
+    )    
+    
     pipe.vae.enable_slicing()
     pipe.vae.enable_tiling()
     torch.cuda.empty_cache()
@@ -41,26 +50,31 @@ def feifeimodload():
 pipe = feifeimodload()
 
 @spaces.GPU()
-def infer(prompt="",  styles_Radio="(None)", Expansion_select=False, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidancescale=3.5, num_feifei=0.35, nsfw_select=False, nsfw_slider=1, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt="",  styles_Radio="(None)", feife_select=False, feife_select= True, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, guidancescale=3.5, num_feifei=0.35, nsfw_select=False, nsfw_slider=1, progress=gr.Progress(track_tqdm=True)):
     
+    Duke86Syl_lora_name=[]
+    adapter_weights_num=[]
+    
+    if feife_select:
+        Duke86Syl_lora_name.append("feifei")
+        adapter_weights_num.append(num_feifei)
+        
+    if bigboobs_select:
+        Duke86Syl_lora_name.append("big-boobs-clothed-v2")
+        adapter_weights_num.append(0.45)
+
     if nsfw_select:
-        pipe.set_adapters(
-            ["feifei", "sldr_flux_nsfw_v2"],
-            adapter_weights=[num_feifei,nsfw_slider],
-        )
-        pipe.fuse_lora(
-            adapter_name=["feifei", "sldr_flux_nsfw_v2"],
-            lora_scale=1.0,
-        )
-    else:
-        pipe.set_adapters(
-            ["feifei"],
-            adapter_weights=[num_feifei],
-        )
-        pipe.fuse_lora(
-            adapter_name=["feifei"],
-            lora_scale=1.0,
-        )
+        Duke86Syl_lora_name.append("sldr_flux_nsfw_v2")
+        adapter_weights_num.append(nsfw_slider)
+
+    pipe.set_adapters(
+        Duke86Syl_lora_name,
+        adapter_weights=adapter_weights_num,
+    )
+    pipe.fuse_lora(
+        adapter_name=Duke86Syl_lora_name,
+        lora_scale=1.0,
+    )
         
     if Expansion_select:
         # Define the replacement string
@@ -159,8 +173,8 @@ with gr.Blocks(css=css) as demo:
         )
         # 定义提交按钮
         feifei_chat_btn = gr.Button(value="Gen Prompt")
-        Expansion_select = gr.Checkbox(label="FeiFei Expansion") 
-        
+        feife_select = gr.Checkbox(label="FeiFei Expansion") 
+        bigboobs_select = gr.Checkbox(label="bigboobs", value=True) 
         with gr.Accordion("Advanced Settings", open=False):
             
             seed = gr.Slider(
@@ -239,7 +253,7 @@ with gr.Blocks(css=css) as demo:
 
     run_button.click(
         fn = infer,
-        inputs = [prompt, styles_Radio, Expansion_select, seed, randomize_seed, width, height, num_inference_steps, guidancescale, num_feifei, nsfw_select, nsfw_slider],
+        inputs = [prompt, styles_Radio, feife_select, bigboobs_select, seed, randomize_seed, width, height, num_inference_steps, guidancescale, num_feifei, nsfw_select, nsfw_slider],
         outputs = [result, seed]
     )
     feifei_chat_btn.click(