Update app.py

app.py

@@ -22,7 +22,7 @@ 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, style="Basic", 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, mode="Basic", progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
@@ -38,11 +38,11 @@ def predict(prompt, upload_images, ip_adapter_scale=0.5, negative_prompt="", see
     
     generator = torch.Generator(device="cuda").manual_seed(seed)
 
-    if style == "Style1":
+    if mode == "Style-Only":
         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":
+    elif mode == "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":
+    elif mode == "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
@@ -212,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, style],
+        inputs=[prompt, files, ip_adapter_scale, negative_prompt, seed, randomize_seed, center_crop, width, height, guidance_scale, num_inference_steps, mode],
         outputs=[result, seed]
     )