Merge pull request #41 from code-kern-ai/remove-gates

Removes gates

Author: JWittmeyer

src/components/models-download/AddModelDownloadModal.tsx

@@ -25,7 +25,6 @@ export default function AddModelDownloadModal() {
     const [modelName, setModelName] = useState('');
     const [colorDownloadedModels, setColorDownloadedModels] = useState<boolean[]>([]);
     const [hoverBoxList, setHoverBoxList] = useState<any[]>([]);
-    const [lineSeparatorIndex, setLineSeparatorIndex] = useState(-1);
     const [filteredList, setFilteredList] = useState<any[]>(null);
 
     useEffect(() => {
@@ -49,7 +48,6 @@ export default function AddModelDownloadModal() {
             }
         });
         setHoverBoxList(hoverBoxList);
-        setLineSeparatorIndex(filteredList.findIndex((model: any) => !model.description));
     }, [modelsDownloaded, filteredList, modalAddModel, modelsList, modelName]);
 
     const addModel = useCallback(() => {
@@ -83,7 +81,7 @@ export default function AddModelDownloadModal() {
                     hasSearchBar={true} dropdownItemsClasses="max-h-96 overflow-y-auto" ignoreDisabledForSearch={true}
                     selectedOption={(option: any) => {
                         setModelName(option.configString);
-                    }} optionsHaveHoverBox={true} hoverBoxList={hoverBoxList} lineSeparatorIndex={lineSeparatorIndex}
+                    }} optionsHaveHoverBox={true} hoverBoxList={hoverBoxList}
                     searchTextTyped={(option: any) => setModelName(option)}
                     filteredOptions={(option) => {
                         setFilteredList(modelsList.filter((model: any) => model.configString.includes(option)));

src/components/models-download/ModelsDownload.tsx

@@ -66,9 +66,6 @@ export default function ModelsDownload() {
                                 <th scope="col"
                                     className="px-3 py-2 text-center text-xs font-medium uppercase tracking-wide text-gray-500">
                                     Name</th>
-                                <th scope="col"
-                                    className="px-3 py-2 text-center text-xs font-medium uppercase tracking-wide text-gray-500">
-                                    Zero-shot</th>
                                 <th scope="col"
                                     className="px-3 py-2 text-center text-xs font-medium uppercase tracking-wide text-gray-500">
                                     Revision</th>
@@ -95,15 +92,6 @@ export default function ModelsDownload() {
                                     <td className="text-center px-3 py-2 text-sm text-gray-500">
                                         <span className="inline-block mr-2">{model.name}</span>
                                     </td>
-                                    <td className="text-center px-3 py-2 text-sm text-gray-500">
-                                        <div className="flex justify-center">
-                                            {model.zeroShotPipeline ? <Tooltip content={TOOLTIPS_DICT.MODELS_DOWNLOAD.USED_ZS} color="invert" placement="top" className="cursor-auto">
-                                                <IconCheckbox className="h-5 w-5"></IconCheckbox>
-                                            </Tooltip> : <Tooltip content={TOOLTIPS_DICT.MODELS_DOWNLOAD.NOT_USED_ZS} color="invert" placement="top" className="cursor-auto">
-                                                <IconBan className="h-5 w-5"></IconBan>
-                                            </Tooltip>}
-                                        </div>
-                                    </td>
                                     <td className="text-center px-3 py-2 text-sm text-gray-500">
                                         {model.revision ? model.revision : '-'}
                                     </td>