Update app.py

app.py

@@ -15,33 +15,33 @@ from huggingface_hub import hf_hub_download, list_repo_files
 # Define constants
 DESCRIPTION = "[UGround Demo](https://osu-nlp-group.github.io/UGround/)"
 _SYSTEM = "You are a very helpful assistant."
-MIN_PIXELS = 256 * 28 * 28
-MAX_PIXELS = 1344 * 1344
+MIN_PIXELS = 802816
+MAX_PIXELS = 1806336
 
 # Specify the model repository and destination folder
 # https://huggingface.co/osunlp/UGround-V1-2B
 model_repo = "osunlp/UGround-V1-2B"
 destination_folder = "./UGround-V1-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}")
+# 
+# # 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}")
 
 model = Qwen2VLForConditionalGeneration.from_pretrained(
-    destination_folder,
+    model_repo,
     torch_dtype=torch.bfloat16,
     device_map="cpu",
 )
 
 # Load the processor
-processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", min_pixels=MIN_PIXELS, max_pixels=MAX_PIXELS)
+processor = AutoProcessor.from_pretrained(model_repo, min_pixels=MIN_PIXELS, max_pixels=MAX_PIXELS)
 
 # Helper functions
 def draw_point(image_input, point=None, radius=5):
@@ -134,7 +134,7 @@ def run_showui(image, query, session_id, iterations=1):
         )[0]
 
         click_xy = ast.literal_eval(output_text)
-        
+
 
         # Draw point on the current image
         result_image = draw_point(image_path, click_xy, radius=10)