Update app.py

README.md

@@ -10,17 +10,8 @@ pinned: true
 license: creativeml-openrail-m
 ---
 
-> A work-in-progress – may be buggy and slow!
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
-
-### Notes on Gradio changes
-- onclick events for canvas? we can draw, but can I get coordinates?
-- checkboxes seem a bit busted with indexes
-- set canvas default to segmentation output, make small edits rather than doing whole thing
-
 ### ToDos:
 - [ ] setting a random seed
 - [ ] click support for segmentation
 - [ ] draw on mask
-- [ ] batching support (blocked by inpainting issue involving tensors + batched inputs)
+- [ ] batching support 

app.py

@@ -13,7 +13,7 @@ import gradio as gr
 from transformers import DetrFeatureExtractor, DetrForSegmentation, DetrConfig
 from transformers.models.detr.feature_extraction_detr import rgb_to_id
 
-from diffusers import StableDiffusionInpaintPipeline, EulerDiscreteScheduler
+from diffusers import StableDiffusionInpaintPipeline, DPMSolverMultistepScheduler
 
 parser = ArgumentParser()
 parser.add_argument('--disable-cuda', action='store_true')
@@ -68,6 +68,7 @@ def clean_mask(mask, max_kernel: int = 23, min_kernel: int = 5):
 
 feature_extractor, segmentation_model, segmentation_cfg = load_segmentation_models()
 pipe = load_diffusion_pipeline()
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
 segmentation_model = segmentation_model.to(device)
 pipe = pipe.to(device)