yasserrmd commited on
Commit
19f0e18
·
verified ·
1 Parent(s): b875cd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -32,6 +32,8 @@ def generate_sketch(prompt, num_inference_steps, guidance_scale):
32
  # if result['label'] == 'nsfw' and result['score'] > NSFW_THRESHOLD:
33
  # return gr.update(visible=False),gr.Text(value="Inappropriate prompt detected. Please try another prompt.")
34
  print(prompt)
 
 
35
  image = pipe("sketched style, " + prompt,
36
  num_inference_steps=num_inference_steps,
37
  guidance_scale=guidance_scale,
 
32
  # if result['label'] == 'nsfw' and result['score'] > NSFW_THRESHOLD:
33
  # return gr.update(visible=False),gr.Text(value="Inappropriate prompt detected. Please try another prompt.")
34
  print(prompt)
35
+ prompt= "sketched style, " + prompt
36
+ prompt= "shou_xin, " + prompt
37
  image = pipe("sketched style, " + prompt,
38
  num_inference_steps=num_inference_steps,
39
  guidance_scale=guidance_scale,