Update app.py

app.py

@@ -21,7 +21,8 @@ pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max
 
 @spaces.GPU(enable_queue=True)
-def predict(prompt, upload_images, ip_adapter_scale=0.5, negative_prompt="", seed=100, randomize_seed=False, center_crop=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=50, progress=gr.Progress(track_tqdm=True)):
+
+def predict(prompt, upload_images, ip_adapter_scale=0.5, negative_prompt="", seed=100, randomize_seed=False, center_crop=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=50, style="Basic", progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
@@ -36,7 +37,19 @@ def predict(prompt, upload_images, ip_adapter_scale=0.5, negative_prompt="", see
     #     ip_adapter_images = [image.resize((224, 224)) for image in ip_adapter_images]
     
     generator = torch.Generator(device="cuda").manual_seed(seed)
-    pipe.set_ip_adapter_scale([ip_adapter_scale])
+
+    if style == "Style1":
+        adapter_scale = {"down": {"block_2": [ip_adapter_scale, 0.0]}, "up": {"block_0": [0.0, ip_adapter_scale, 0.0]}, "mid": ip_adapter_scale}
+    elif style == "Style2":
+        adapter_scale = {"down": {"block_2": [ip_adapter_scale, ip_adapter_scale]}, "up": {"block_0": [0.0, ip_adapter_scale, 0.0]}}
+    elif style == "Style3":
+        adapter_scale = {"down": {"block_2": [ip_adapter_scale, 0.0], "block_1": [0.0, ip_adapter_scale]}, "up": {"block_0": [0.0, ip_adapter_scale, 0.0]}}
+    else:
+        adapter_scale = ip_adapter_scale
+        
+        
+    
+    pipe.set_ip_adapter_scale([adapter_scale])
     
     image = pipe(
         prompt=prompt,
@@ -107,7 +120,7 @@ with gr.Blocks(css=css) as demo:
                         value=1.0,
                     )
 
-                    gr.Dropdown(
+                    style = gr.Dropdown(
                         ["Basic", "Style1", "Style2","Style3"], label="Adapter Type", info="Style Transfer Options"
                     )                    
             with gr.Column():
@@ -199,7 +212,7 @@ with gr.Blocks(css=css) as demo:
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=predict,
-        inputs=[prompt, files, ip_adapter_scale, negative_prompt, seed, randomize_seed, center_crop, width, height, guidance_scale, num_inference_steps],
+        inputs=[prompt, files, ip_adapter_scale, negative_prompt, seed, randomize_seed, center_crop, width, height, guidance_scale, num_inference_steps, style],
         outputs=[result, seed]
     )