Update app.py

app.py

@@ -11,7 +11,7 @@ import ast
 import os
 from datetime import datetime
 import numpy as np
-from huggingface_hub import snapshot_download
+from huggingface_hub import hf_hub_download, list_repo_files
 
 # Define constants
 DESCRIPTION = "[ShowUI Demo](https://huggingface.co/showlab/ShowUI-2B)"
@@ -29,7 +29,20 @@ 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")
+# Specify the model repository and destination folder
+model_repo = "showlab/ShowUI-2B"
+destination_folder = "./showui-2b"
+
+# Ensure the destination folder exists
+os.makedirs(destination_folder, exist_ok=True)
+
+# List all files in the repository
+files = list_repo_files(repo_id=model_repo)
+
+# Download each file to the destination folder
+for file in files:
+    file_path = hf_hub_download(repo_id=model_repo, filename=file, local_dir=destination_folder)
+    print(f"Downloaded {file} to {file_path}")
 
 # Helper functions
 def draw_point(image_input, point=None, radius=5):