Update app.py

app.py

@@ -11,6 +11,7 @@ import ast
 import os
 from datetime import datetime
 import numpy as np
+from huggingface_hub import snapshot_download
 
 # Define constants
 DESCRIPTION = "[ShowUI Demo](https://huggingface.co/showlab/ShowUI-2B)"
@@ -28,6 +29,8 @@ model = Qwen2VLForConditionalGeneration.from_pretrained(
 # Load the processor
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", min_pixels=MIN_PIXELS, max_pixels=MAX_PIXELS)
 
+local_dir = snapshot_download(repo_id="showlab/ShowUI-2B", local_dir="./showui-2b")
+
 # Helper functions
 def draw_point(image_input, point=None, radius=5):
     """Draw a point on the image."""