Update app.py

app.py

@@ -34,12 +34,23 @@ def generate_sketch(prompt,style, num_inference_steps, guidance_scale):
     print(prompt)
     
 
-    if style=='shou_xin':
-        prompt= "shou_xin, " + prompt
+   match style:
+    case 'shou_xin':
+        prompt = "shou_xin, " + prompt
         pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
-    else:
-        prompt= "sketched style, " + prompt
+    case 'sketched':
+        prompt = "sketched style, " + prompt
         pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch", weight_name="FLUX-dev-lora-children-simple-sketch.safetensors")
+    case 'sketch_paint':
+        prompt = "Sketch paint, " + prompt
+        pipe.load_lora_weights("strangerzonehf/Sketch-Paint", weight_name="Sketch-Paint.safetensors")
+    case 'sketch_sized':
+        prompt = "Sketch Sized, " + prompt
+        pipe.load_lora_weights("strangerzonehf/Flux-Sketch-Sized-LoRA", weight_name="Flux-Sketch-Sized-LoRA.safetensors")
+    case _:
+        prompt = "shou_xin, " + prompt
+        pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
+
     
     pipe.fuse_lora(lora_scale=1.5)
     
@@ -71,7 +82,7 @@ interface = gr.Interface(
     inputs=[
         "text",  # Prompt input
         gr.Dropdown(
-        ["sketched", "shou_xin"],value='sketched', label="Style"
+        ["sketched", "shou_xin","Sketch-Paint","Flux-Sketch-Sized-LoRA"],value='sketched', label="Style"
         ),
         gr.Slider(5, 50, value=24, step=1, label="Number of Inference Steps"),  # Slider for num_inference_steps
         gr.Slider(1.0, 10.0, value=3.5, step=0.1, label="Guidance Scale")  # Slider for guidance_scale