Skip to content

Commit cfcdedb

Browse files
committed
fix: models hub should show latest data only
1 parent 4d4cf89 commit cfcdedb

File tree

2 files changed

+14
-12
lines changed

2 files changed

+14
-12
lines changed

web-app/src/hooks/useModelSources.ts

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -31,15 +31,9 @@ export const useModelSources = create<ModelSourcesState>()(
3131
set({ loading: true, error: null })
3232
try {
3333
const newSources = await fetchModelCatalog()
34-
const currentSources = get().sources
3534

3635
set({
37-
sources: [
38-
...newSources,
39-
...currentSources.filter(
40-
(e) => !newSources.some((s) => s.model_name === e.model_name)
41-
),
42-
],
36+
sources: newSources.length ? newSources : get().sources,
4337
loading: false,
4438
})
4539
} catch (error) {

web-app/src/routes/hub/index.tsx

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -145,8 +145,12 @@ function Hub() {
145145
const repoInfo = await fetchHuggingFaceRepo(search.repo)
146146
if (repoInfo) {
147147
const catalogModel = convertHfRepoToCatalogModel(repoInfo)
148-
setHuggingFaceRepo(catalogModel)
149-
addSource(catalogModel)
148+
if (
149+
!sources.some((s) => s.model_name === catalogModel.model_name)
150+
) {
151+
setHuggingFaceRepo(catalogModel)
152+
addSource(catalogModel)
153+
}
150154
}
151155

152156
await fetchSources()
@@ -157,7 +161,7 @@ function Hub() {
157161
}
158162
}, 500)
159163
}
160-
}, [convertHfRepoToCatalogModel, fetchSources, addSource, search])
164+
}, [convertHfRepoToCatalogModel, fetchSources, addSource, search, sources])
161165

162166
// Sorting functionality
163167
const sortedModels = useMemo(() => {
@@ -247,8 +251,12 @@ function Hub() {
247251
const repoInfo = await fetchHuggingFaceRepo(e.target.value)
248252
if (repoInfo) {
249253
const catalogModel = convertHfRepoToCatalogModel(repoInfo)
250-
setHuggingFaceRepo(catalogModel)
251-
addSource(catalogModel)
254+
if (
255+
!sources.some((s) => s.model_name === catalogModel.model_name)
256+
) {
257+
setHuggingFaceRepo(catalogModel)
258+
addSource(catalogModel)
259+
}
252260
}
253261

254262
// Original addSource logic (if needed)

0 commit comments

Comments
 (0)