Update app.py

app.py

@@ -18,7 +18,8 @@ model_id = "google/gemma-3-4b-pt"
 
 MODEL_OPTIONS = {
     "🔵 Fine-tuned": "sergiopaniego/gemma-3-4b-pt-object-detection",
-    "🟣 Fine-tuned (updated tokenizer with `<loc>` tokens)": "sergiopaniego/gemma-3-4b-pt-object-detection-loc-tokens",
+    "🟣 Fine-tuned(updated tokenizer with `<loc>` tokens)": "sergiopaniego/gemma-3-4b-pt-object-detection-loc-tokens",
+    "✨ Fine-tuned+": "ajaymin28/gemma-3-4b-pt-obj-det-loc-tok-1024-single-stage",
 }
 
 resize_size = 512 if "SmolVLM" in model_id else 896
@@ -99,6 +100,7 @@ with gr.Blocks(theme=Soft(), css=css_hide_share) as demo:
 This demo compares two fine-tuned versions of **Gemma 3 (4B)** for object detection:
 - **🔵 Fine-tuned for object detection**: trained to predict bounding boxes and class labels using the original tokenizer.
 - **🟣 Fine-tuned (updated tokenizer with `<loc>` tokens)**: same task, but uses a tokenizer updated to better encode spatial information through `<locYYYY>` tokens.
+- **✨ Fine-tuned+ (updated tokenizer with `<loc>` tokens)**: Same as `<locYYYY>` tokens finetuning but single stage training.
 Select a model and click **Generate** to visualize its prediction on a random test image.
 """)