Update app.py

app.py

@@ -24,12 +24,17 @@ from modelscope.hub.snapshot_download import snapshot_download as ms_snapshot_do
 import spaces
 
 
-
-sd_dir = ms_snapshot_download('xichen/cleansd', cache_dir='./modelscope')
+from huggingface_hub import snapshot_download
+snapshot_download(repo_id="xichenhku/cleansd", local_dir="./cleansd")
 print('=== Pretrained SD weights downloaded ===')
-model_dir = ms_snapshot_download('xichen/MimicBrush', cache_dir='./modelscope')
+snapshot_download(repo_id="xichenhku/MimicBrush", local_dir="./MimicBrush")
 print('=== MimicBrush weights downloaded ===')
 
+#sd_dir = ms_snapshot_download('xichen/cleansd', cache_dir='./modelscope')
+#print('=== Pretrained SD weights downloaded ===')
+#model_dir = ms_snapshot_download('xichen/MimicBrush', cache_dir='./modelscope')
+#print('=== MimicBrush weights downloaded ===')
+
 val_configs = OmegaConf.load('./configs/inference.yaml')
 
 # === import Depth Anything ===
@@ -309,7 +314,7 @@ with gr.Blocks() as demo:
  with gr.Row():
  base = gr.ImageEditor( label="Source",
  type="pil",
- brush=gr.Brush(colors=["#000000"],default_size = 50,color_mode = "fixed"),
+ brush=gr.Brush(colors=["#000000"],default_size = 30,color_mode = "fixed"),
  layers = False,
  interactive=True
  )