add base_model field

prisma/schema.prisma

@@ -22,6 +22,7 @@ model Model {
   user            User?     @relation(fields: [userId], references: [sub])
   userId          String?
   metadata        String?
+  base_model      String?
   gallery         Gallery[]
   comments        Comment[]
 }

src/lib/components/models/drawer/Drawer.svelte

@@ -147,7 +147,7 @@
               </div>
               <div class="px-4 py-3">
                 <p class="text-neutral-400 uppercase text-[10px]">base_model</p>
-                <p class="text-neutral-100 font-medium text-base mt-0.5">{model?.infos?.base_model ?? 'N/A'}</p>
+                <p class="text-neutral-100 font-medium text-base mt-0.5">{model?.base_model ?? 'N/A'}</p>
               </div>
             </div>
           {/if}

src/lib/type.ts

@@ -24,6 +24,7 @@ export interface ModelCard {
   image: string,
   instance_prompt?: string,
   metadata?: string,
+  base_model?: string,
   isPublic: boolean,
   gallery?: CommunityCard[],
   comments?: CommentType[],

src/routes/api/models/+server.ts

@@ -89,6 +89,8 @@ export async function PATCH({ request } : RequestEvent) {
 
     }
 
+    const base_model = hugging_face_model.tags.find((tag: string) => tag.startsWith("base_model:"))?.split(":")[1] ?? null
+
     if (!hugging_face_model?.[0]) {
       continue;
     }
@@ -104,6 +106,10 @@ export async function PATCH({ request } : RequestEvent) {
         ...(hugging_face_model2?.cardData?.instance_prompt ? {
           instance_prompt: hugging_face_model2?.cardData?.instance_prompt,
         } : {}
+        ),
+        ...(base_model ? {
+          base_model: base_model
+        } : {}
         )
       }
     })

src/routes/api/models/[id]/+server.ts

@@ -25,6 +25,7 @@ export async function GET({ url, params } : RequestEvent) {
       image: true,
       instance_prompt: true,
       metadata: true,
+      base_model: true,
       gallery: {
         select: {
           id: true,
@@ -179,13 +180,19 @@ export async function POST({ params, cookies, fetch } : RequestEvent) {
   const data = await fetch(`https://huggingface.co/api/models/${model.id}`)
   const hf_model = await data.json()
 
+  const base_model = hf_model.tags.find((tag: string) => tag.startsWith("base_model:"))?.split(":")[1] ?? null
+
   await prisma.model.update({
     where: {
       id,
     },
     data: {
       isPublic: true,
-      instance_prompt: hf_model?.cardData?.instance_prompt
+      instance_prompt: hf_model?.cardData?.instance_prompt,
+      ...(base_model ? {
+        base_model: base_model
+      } : {}
+      )
     }
   })
 

src/routes/api/scrap-models/+server.ts

@@ -58,6 +58,8 @@ export async function POST({ request }) {
       }
     }
 
+    const base_model = model.tags.find((tag: string) => tag.startsWith("base_model:"))?.split(":")[1] ?? null
+
     await prisma.model.create({
       data: {
         id: model.id,
@@ -66,6 +68,10 @@ export async function POST({ request }) {
         downloads: model.downloads,
         createdAt: model.createdAt,
         isPublic: false,
+        ...(base_model ? {
+          base_model: base_model
+        } : {}
+        )
       }
     })
     .then(() => {