Merge pull request #20 from Stability-AI/stablewaheed

Forcing white inpaint brush
Stability-AI · Feb 8, 2025 · 03aecf4 · 03aecf4
2 parents 7aeaee7 + b4fe28a
commit 03aecf4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sd35-inpainting-gradio/app.py b/sd35-inpainting-gradio/app.py
@@ -65,7 +65,7 @@ def _predict(self, mask, prompt, progress=gr.Progress(track_tqdm=True)):
         return images[0]
 
     def _start_gradio(self):
-        white_brush = gr.Brush(colors=['#FFFFFF'], color_mode='fixed')
+        white_brush = gr.Brush(default_color='#FFFFFF', colors=['#FFFFFF'], color_mode='fixed')
 
         gr.Interface(
             self._predict,