From 68fe4cc9dc1389558f9e9415b6d6c13b21dddbfa Mon Sep 17 00:00:00 2001
From: sumona-banerjeee <sumobanerjee2000@gmail.com>
Date: Fri, 13 Mar 2026 18:21:03 +0530
Subject: [PATCH] Feat: Added extra drop down button for yoloe with vl

---
 app/upload/page.tsx           | 7 +++++--
 components/upload-section.tsx | 5 +++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/app/upload/page.tsx b/app/upload/page.tsx
index 837275e..6c60ee0 100644
--- a/app/upload/page.tsx
+++ b/app/upload/page.tsx
@@ -317,9 +317,12 @@ export default function UploadPage() {
                                             <SelectItem value="sam3">
                                                 <span className="font-medium">OpenAI SAM 3 Segmentation Model</span>
                                             </SelectItem>
-                                            {/* <SelectItem value="yoloe">
+                                            <SelectItem value="yoloe">
                                                 <span className="font-medium">YOLOE Open-Vocabulary Detection</span>
-                                            </SelectItem> */}
+                                            </SelectItem>
+                                            <SelectItem value="yoloe_trained_vl">
+                                                <span className="font-medium">YOLOE With Vision Language Model</span>
+                                            </SelectItem>
                                         </SelectContent>
                                     </Select>
                                 </div>
diff --git a/components/upload-section.tsx b/components/upload-section.tsx
index eb1993e..bad0667 100644
--- a/components/upload-section.tsx
+++ b/components/upload-section.tsx
@@ -339,6 +339,11 @@ export function UploadSection({ onDetectionComplete, onDetectionTypeChange }: Up
                     <span>YOLOE</span>
                   </div>
                 </SelectItem>
+                <SelectItem value="yoloe_trained_vl">
+                  <div className="flex items-center gap-2">
+                    <span>YOLOE Trained</span>
+                  </div>
+                </SelectItem>
               </SelectContent>
             </Select>
           </div>