From 8afaf55eb5d7d83ab5f1e8ca7a2d2a3963f0d4ac Mon Sep 17 00:00:00 2001
From: Manan17 <shahmanan170602@gmail.com>
Date: Fri, 6 Mar 2026 08:47:56 +0000
Subject: [PATCH] derive effective model type from isVisionModel for dataset
 search filterin

---
 .../frontend/src/features/studio/sections/dataset-section.tsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/studio/frontend/src/features/studio/sections/dataset-section.tsx b/studio/frontend/src/features/studio/sections/dataset-section.tsx
index 5af9900e3a..8438e1a7fb 100644
--- a/studio/frontend/src/features/studio/sections/dataset-section.tsx
+++ b/studio/frontend/src/features/studio/sections/dataset-section.tsx
@@ -77,6 +77,7 @@ export function DatasetSection() {
     hfToken,
     modelType,
     isVisionModel,
+    isCheckingVision,
     datasetSliceStart,
     setDatasetSliceStart,
     datasetSliceEnd,
@@ -96,6 +97,7 @@ export function DatasetSection() {
       hfToken: s.hfToken,
       modelType: s.modelType,
       isVisionModel: s.isVisionModel,
+      isCheckingVision: s.isCheckingVision,
       datasetSliceStart: s.datasetSliceStart,
       setDatasetSliceStart: s.setDatasetSliceStart,
       datasetSliceEnd: s.datasetSliceEnd,
@@ -122,7 +124,7 @@ export function DatasetSection() {
     setInputValue(val);
   }
 
-  const effectiveModelType = isVisionModel ? "vision" : modelType;
+  const effectiveModelType = !isCheckingVision && isVisionModel ? "vision" : modelType;
 
   const {
     results: hfResults,