From cb3f3f4d0c43390dc318df8a1ae3e6a732cfb6fe Mon Sep 17 00:00:00 2001
From: Manan17 <shahmanan170602@gmail.com>
Date: Fri, 6 Mar 2026 07:44:57 +0000
Subject: [PATCH] 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,
   });