update trending score

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

@@ -81,7 +81,7 @@ export async function PATCH({ request } : RequestEvent) {
     const hugging_face_request = await fetch(`https://huggingface.co/api/models?id=${model.id}&sort=likes7d`)
     const hugging_face_model = await hugging_face_request.json()?.catch(() => {})
 
-    if (!hugging_face_model) {
+    if (!hugging_face_model?.[0]) {
       continue;
     }
     await prisma.model.update({
@@ -89,10 +89,10 @@ export async function PATCH({ request } : RequestEvent) {
         id: model.id
       },
       data: {
-        likes: hugging_face_model.likes,
-        downloads: hugging_face_model.downloads,
-        likes7d: hugging_face_model.trendingScore,
-        id: hugging_face_model.id,
+        likes: hugging_face_model?.[0]?.likes,
+        downloads: hugging_face_model?.[0]?.downloads,
+        likes7d: hugging_face_model?.[0]?.trendingScore,
+        id: hugging_face_model?.[0]?.id,
       }
     })
     .then(() => {

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

@@ -124,7 +124,7 @@ export async function POST({ params, cookies, fetch } : RequestEvent) {
     }, { status: 404 })
   }
 
-  const data = await fetch(`https://huggingface.co/api/models?id=${model.id}&sort=likes7d`)
+  const data = await fetch(`https://huggingface.co/api/models/${model.id}`)
   const hf_model = await data.json()
 
   await prisma.model.update({
@@ -133,10 +133,6 @@ export async function POST({ params, cookies, fetch } : RequestEvent) {
     },
     data: {
       isPublic: true,
-      likes: hf_model?.likes,
-      downloads: hf_model?.downloads,
-      likes7d: hf_model?.trendingScore,
-      id: hf_model?.id,
       instance_prompt: hf_model?.cardData?.instance_prompt
     }
   })

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

@@ -30,7 +30,7 @@ export async function POST({ request, fetch, cookies }) {
   const res = await fetch(`https://huggingface.co/api/models?id=${model.id}&sort=likes7d`)
   const data = await res.json();
 
-  if (data?.error) {
+  if (data?.error || !data?.[0]) {
     return json({
       error: {
         id: "Model not found on Hugging Face"
@@ -56,9 +56,9 @@ export async function POST({ request, fetch, cookies }) {
     data: {
       id: model.id,
       image: model.image,
-      likes: data.likes,
-      downloads: data.downloads,
-      likes7d: data.trendingScore,
+      likes: data?.[0]?.likes,
+      downloads: data?.[0]?.downloads,
+      likes7d: data?.[0]?.trendingScore,
       isPublic: false,
     }
   })

src/routes/api/scrap-models/[slug]/+server.ts

@@ -91,6 +91,10 @@ export async function PATCH({ request, params } : RequestEvent) {
   for (const model of models) {
     const hf_model = await fetch(`https://huggingface.co/api/models?id=${model.id}&sort=likes7d`)
     const new_model = await hf_model.json();
+
+    if (!new_model?.[0]) {
+      continue;
+    }
     const update_data = {
       instance_prompt: model.instance_prompt,
       image: model.image,
@@ -114,10 +118,10 @@ export async function PATCH({ request, params } : RequestEvent) {
       },
       data: {
         ...update_data,
-        likes: new_model.likes,
-        downloads: new_model.downloads,
-        likes7d: new_model.trendingScore,
-        id: new_model.id,
+        likes: new_model?.[0]?.likes,
+        downloads: new_model?.[0]?.downloads,
+        likes7d: new_model?.[0]?.trendingScore,
+        id: new_model?.[0]?.id,
       }
     })
   }