yasserrmd commited on
Commit
a59bcf0
1 Parent(s): 7839bee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ pipe.to("cuda")
12
  # Define the function to generate the sketch
13
  @spaces.GPU
14
  def generate_sketch(prompt, num_inference_steps, guidance_scale):
15
- image = pipe("sketched style," + prompt,
16
  num_inference_steps=num_inference_steps,
17
  guidance_scale=guidance_scale,
18
  ).images[0]
 
12
  # Define the function to generate the sketch
13
  @spaces.GPU
14
  def generate_sketch(prompt, num_inference_steps, guidance_scale):
15
+ image = pipe("sketched style, " + prompt,
16
  num_inference_steps=num_inference_steps,
17
  guidance_scale=guidance_scale,
18
  ).images[0]