Update app.py

app.py

@@ -19,16 +19,6 @@ _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
 
-# Load the model
-model = Qwen2VLForConditionalGeneration.from_pretrained(
-    "./showui-2b",
-    torch_dtype=torch.bfloat16,
-    device_map="auto",
-)
-
-# Load the processor
-processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", min_pixels=MIN_PIXELS, max_pixels=MAX_PIXELS)
-
 # Specify the model repository and destination folder
 model_repo = "showlab/ShowUI-2B"
 destination_folder = "./showui-2b"
@@ -43,6 +33,16 @@ files = list_repo_files(repo_id=model_repo)
 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}")
+    
+# Load the model
+model = Qwen2VLForConditionalGeneration.from_pretrained(
+    "./showui-2b",
+    torch_dtype=torch.bfloat16,
+    device_map="auto",
+)
+
+# Load the processor
+processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", min_pixels=MIN_PIXELS, max_pixels=MAX_PIXELS)
 
 # Helper functions
 def draw_point(image_input, point=None, radius=5):