Update app.py

app.py

@@ -19,24 +19,24 @@ _SYSTEM = "Based on the screenshot of the page, I give a text description and yo
 MIN_PIXELS = 256 * 28 * 28
 MAX_PIXELS = 1344 * 28 * 28
 
-# # Specify the model repository and destination folder
-# model_repo = "showlab/ShowUI-2B"
-# destination_folder = "./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)
+# 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)
+# 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}")
+# 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}")
 
 model = Qwen2VLForConditionalGeneration.from_pretrained(
-    # "./showui-2b",
-    "showlab/ShowUI-2B",
+    "./showui-2b",
+    # "showlab/ShowUI-2B",
     torch_dtype=torch.bfloat16,
     device_map="cpu",
 )