diff --git a/app/components/Models.tsx b/app/components/Models.tsx index 74f096e..dee935e 100644 --- a/app/components/Models.tsx +++ b/app/components/Models.tsx @@ -174,18 +174,9 @@ const Models: React.FC = () => { const [currentCategory, setCurrentCategory] = useState(localStorage.getItem("activeCategory")); useEffect(() => { - if (!localStorage.getItem('selectedModel')) { - localStorage.setItem("selectedModel", "Offline Fast" ) - } - - if (!localStorage.getItem("activeCategory")) { - setActiveCategory('Code') - localStorage.setItem('activeCategory' ,'Code') - } - - if (!localStorage.getItem("model")) { - localStorage.setItem("model" ,'starcoder2') - } + // Retrieve initial values from localStorage + setRadioSelection(localStorage.getItem('radioSelection') || ''); + setSelectedModel(localStorage.getItem('selectedModel') || ''); const handleStorageChange = () => { setSelectedModel(localStorage.getItem('selectedModel') || ''); @@ -194,8 +185,6 @@ const Models: React.FC = () => { // Update immediately when localStorage changes window.addEventListener('storage', handleStorageChange); - setRadioSelection(localStorage.getItem('radioSelection') || ''); - setSelectedModel(localStorage.getItem('selectedModel') || ''); // Cleanup listener on component unmount return () => { window.removeEventListener('storage', handleStorageChange);