Refactor constants

app.py

@@ -18,6 +18,7 @@ from transformers import (
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 
+# Handle spaces.GPU decorator for GPU allocation in Spaces
 if os.environ.get("SPACES_ZERO_GPU") is not None:
     import spaces
 else:
@@ -31,10 +32,10 @@ else:
             return wrapper
 
 
+# Define constants
 EXAMPLES_DIR = Path(__file__).parent / "examples"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-
-model_ids = [
+MODEL_IDS = [
     "Qwen/Qwen2-VL-2B-Instruct",  # https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct
     "Qwen/Qwen2-VL-7B-Instruct",  # https://huggingface.co/Qwen/Qwen2-VL-7B-Instruct
     "Qwen/Qwen2.5-VL-3B-Instruct",  # https://huggingface.co/Qwen/Qwen2.5-VL-3B-Instruct
@@ -88,7 +89,7 @@ with gr.Blocks() as demo:
             gr.Markdown("## Settings")
 
             input_model_id = gr.Dropdown(
-                choices=model_ids,
+                choices=MODEL_IDS,
                 label="✨ Select Model ID",
             )
             default_system_prompt = 'You are a helpful assistant to detect objects in images. When asked to detect elements based on a description, you return a valid JSON object containing bounding boxes for all elements in the form `[{"bbox_2d": [xmin, ymin, xmax, ymax], "label": "placeholder"}, ...]`. For example, a valid response could be: `[{"bbox_2d": [10, 30, 20, 60], "label": "placeholder"}, {"bbox_2d": [40, 15, 52, 27], "label": "placeholder"}]`.'