yasserrmd commited on
Commit
e1df300
1 Parent(s): 01333b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -16
app.py CHANGED
@@ -34,22 +34,22 @@ def generate_sketch(prompt,style, num_inference_steps, guidance_scale):
34
  print(prompt)
35
 
36
 
37
- match style:
38
- case 'shou_xin':
39
- prompt = "shou_xin, " + prompt
40
- pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
41
- case 'sketched':
42
- prompt = "sketched style, " + prompt
43
- pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch", weight_name="FLUX-dev-lora-children-simple-sketch.safetensors")
44
- case 'sketch_paint':
45
- prompt = "Sketch paint, " + prompt
46
- pipe.load_lora_weights("strangerzonehf/Sketch-Paint", weight_name="Sketch-Paint.safetensors")
47
- case 'sketch_sized':
48
- prompt = "Sketch Sized, " + prompt
49
- pipe.load_lora_weights("strangerzonehf/Flux-Sketch-Sized-LoRA", weight_name="Flux-Sketch-Sized-LoRA.safetensors")
50
- case _:
51
- prompt = "shou_xin, " + prompt
52
- pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
53
 
54
 
55
  pipe.fuse_lora(lora_scale=1.5)
 
34
  print(prompt)
35
 
36
 
37
+ match style:
38
+ case 'shou_xin':
39
+ prompt = "shou_xin, " + prompt
40
+ pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
41
+ case 'sketched':
42
+ prompt = "sketched style, " + prompt
43
+ pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch", weight_name="FLUX-dev-lora-children-simple-sketch.safetensors")
44
+ case 'sketch_paint':
45
+ prompt = "Sketch paint, " + prompt
46
+ pipe.load_lora_weights("strangerzonehf/Sketch-Paint", weight_name="Sketch-Paint.safetensors")
47
+ case 'sketch_sized':
48
+ prompt = "Sketch Sized, " + prompt
49
+ pipe.load_lora_weights("strangerzonehf/Flux-Sketch-Sized-LoRA", weight_name="Flux-Sketch-Sized-LoRA.safetensors")
50
+ case _:
51
+ prompt = "shou_xin, " + prompt
52
+ pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
53
 
54
 
55
  pipe.fuse_lora(lora_scale=1.5)