Update app.py

app.py

@@ -27,18 +27,18 @@ MAX_IMAGE_SIZE = 1024
 
 def infer(prompt, negative_prompt, seed, randomize_seed, image_style,cfg,lora_scale):
     pipe.unload_lora_weights()
-    if image_style == "No LoRA":
+    if image_style == "No style":
         pipe.unload_lora_weights()
         
-    elif image_style == "LoRA_detailed":
+    elif image_style == "style_detailed":
         pipe.load_lora_weights("ohkarim/LoRA_logos", weight_name="OH_logos.safetensors", adapter_name="OH_logos")
         prompt=prompt+", detailed, close up, unique background color, OH_logos"
         
-    elif image_style == "LoRA_blacknwhite":
+    elif image_style == "style_blacknwhite":
         pipe.load_lora_weights("ohkarim/lora_logo_blacknwhite", weight_name="bel_blacknwhite_lora.safetensors", adapter_name="bel_blacknwhite")
         prompt=prompt+", black and white, balck and white, minimalist, unique background color, bel_blacknwhite"
 
-    elif image_style == "LoRA_cartoon":
+    elif image_style == "style_cartoon":
         pipe.load_lora_weights("ohkarim/lora_logo_modern", weight_name="lora_modern.safetensors", adapter_name="oh_bel_modern")
         prompt=prompt+", simple, modern, unique background color, oh_bel_modern"
 
@@ -108,7 +108,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             text_prompt = gr.Textbox(label="Input Prompt", placeholder="Example: logo of a coffee shop, cup of coffee, mug, brown shades ", lines=2)
             negative_prompt = gr.Textbox(label="Negative Prompt (optional)", placeholder="Example: blurry, unfocused, complicated", lines=2)
-            image_style = gr.Dropdown(label="Choose a style", choices=["No LoRA", "LoRA_detailed", "LoRA_blacknwhite", "LoRA_cartoon"], value="No LoRA")
+            image_style = gr.Dropdown(label="Choose a style", choices=["No style", "style_detailed", "style_blacknwhite", "style_cartoon"], value="No LoRA")
 
         with gr.Column():
             cfg = gr.Slider(label="Prompt guidance", minimum=1, maximum=20, step=1, value=4)