Update app.py

app.py

@@ -33,16 +33,22 @@ 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}")
+
+@spaces.GPU
+def get_model_processor():
+    # Load the model
+    model = Qwen2VLForConditionalGeneration.from_pretrained(
+        "./showui-2b",
+        torch_dtype=torch.bfloat16,
+        device_map="auto",
+    )
     
-# 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)
+    # Load the processor
+    processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", min_pixels=MIN_PIXELS, max_pixels=MAX_PIXELS)
+
+    return model, proecessor
+
+model, processor = get_model_processor()
 
 # Helper functions
 def draw_point(image_input, point=None, radius=5):