Update app.py

app.py

@@ -19,23 +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}")
 
 global model = Qwen2VLForConditionalGeneration.from_pretrained(
-    "./showui-2b",
+    # "./showui-2b",
+    "showlab/ShowUI-2B",
     torch_dtype=torch.bfloat16,
     device_map="cpu",
 )