Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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]
|