Update app.py

app.py

@@ -11,7 +11,7 @@ device = torch.device('cuda')
 
 controlnet = ControlNetModel.from_pretrained("BlockDetail/PartialSketchControlNet", torch_dtype=torch.float16).to(device)
 pipe = CustomStableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5",
+    "Lykon/dreamshaper-7",
     controlnet=controlnet, torch_dtype=torch.float16
 ).to(device)
 pipe.safety_checker = None
@@ -102,7 +102,7 @@ with gr.Blocks() as demo:
         else:
             dilation_mask = None
         
-        images, new_images = sketch(curr_sketch_image, dilation_mask, prompt, seed, num_steps = 40, dilation = dilation)
+        images, new_images = sketch(curr_sketch_image, dilation_mask, prompt, seed, num_steps = 8, dilation = dilation)
 
         save_sketch = np.array(Image.fromarray(curr_sketch).convert("RGBA"))
         save_sketch[:, :, 3][save_sketch[:, :, 0] > 128] = 0