From fcd52d9bd0fc67c56d8d4045cfa6f1551658a338 Mon Sep 17 00:00:00 2001
From: Manan17 <shahmanan170602@gmail.com>
Date: Fri, 6 Mar 2026 07:44:57 +0000
Subject: [PATCH 1/2] fixing update model type

---
 .../src/features/studio/sections/dataset-section.tsx       | 7 ++++++-
 1 file changed, 6 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 28254adcc7..5af9900e3a 100644
--- a/studio/frontend/src/features/studio/sections/dataset-section.tsx
+++ b/studio/frontend/src/features/studio/sections/dataset-section.tsx
@@ -76,6 +76,7 @@ export function DatasetSection() {
     setDatasetEvalSplit,
     hfToken,
     modelType,
+    isVisionModel,
     datasetSliceStart,
     setDatasetSliceStart,
     datasetSliceEnd,
@@ -94,6 +95,7 @@ export function DatasetSection() {
       setDatasetEvalSplit: s.setDatasetEvalSplit,
       hfToken: s.hfToken,
       modelType: s.modelType,
+      isVisionModel: s.isVisionModel,
       datasetSliceStart: s.datasetSliceStart,
       setDatasetSliceStart: s.setDatasetSliceStart,
       datasetSliceEnd: s.datasetSliceEnd,
@@ -119,6 +121,9 @@ export function DatasetSection() {
     }
     setInputValue(val);
   }
+
+  const effectiveModelType = isVisionModel ? "vision" : modelType;
+
   const {
     results: hfResults,
     isLoading,
@@ -126,7 +131,7 @@ export function DatasetSection() {
     fetchMore,
     error: hfSearchError,
   } = useHfDatasetSearch(debouncedQuery, {
-    modelType,
+    modelType: effectiveModelType,
     accessToken: hfToken || undefined,
   });
 

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 2/2] 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,