change updated models routes

package-lock.json

@@ -9,7 +9,7 @@
 			"version": "0.0.1",
 			"dependencies": {
 				"@aws-sdk/client-s3": "^3.490.0",
-				"@huggingface/hub": "^0.12.3-oauth",
+				"@huggingface/hub": "^0.12.3",
 				"@iconify/svelte": "^3.1.4",
 				"@prisma/client": "^5.7.1",
 				"@svelte-put/clickoutside": "^3.0.1",
@@ -56,6 +56,29 @@
 				"vite": "^4.4.2"
 			}
 		},
+		"../@huggingface-mini-header": {
+			"name": "@huggingface/space-header",
+			"version": "1.0.0",
+			"extraneous": true,
+			"license": "ISC",
+			"devDependencies": {
+				"tsup": "^8.1.0",
+				"typescript": "^5.4.5"
+			}
+		},
+		"../huggingface.js/packages/space-header": {
+			"name": "@huggingface/space-header",
+			"version": "1.0.0",
+			"extraneous": true,
+			"license": "ISC",
+			"devDependencies": {
+				"tsup": "^8.1.0",
+				"typescript": "^5.4.5"
+			}
+		},
+		"../huggingface.js/space-header": {
+			"extraneous": true
+		},
 		"node_modules/@aashutoshrathi/word-wrap": {
 			"version": "1.2.6",
 			"resolved": "https://registry.npmjs.org/@aashutoshrathi/word-wrap/-/word-wrap-1.2.6.tgz",
@@ -1254,9 +1277,9 @@
 			"dev": true
 		},
 		"node_modules/@huggingface/hub": {
-			"version": "0.12.3-oauth",
-			"resolved": "https://registry.npmjs.org/@huggingface/hub/-/hub-0.12.3-oauth.tgz",
-			"integrity": "sha512-9LE4Ded2VsjQTK/rxsIF/nvhkxs1UE+lcpNpuBxnTGNpaVzbO6HIlAtbThgNenKPplRt1iDkn82AQYBco1QHug==",
+			"version": "0.12.3",
+			"resolved": "https://registry.npmjs.org/@huggingface/hub/-/hub-0.12.3.tgz",
+			"integrity": "sha512-QvK7YSKktrNu/Vyr/oAL2xcCwCiNcf4hPfzra6/jtNNK0GAZXwqOYwB+Kt2hil/XShh1e5cGWDTyKPU5K7zCUA==",
 			"dependencies": {
 				"hash-wasm": "^4.9.0"
 			},

package.json

@@ -42,7 +42,7 @@
 	"type": "module",
 	"dependencies": {
 		"@aws-sdk/client-s3": "^3.490.0",
-		"@huggingface/hub": "^0.12.3-oauth",
+		"@huggingface/hub": "^0.12.3",
 		"@iconify/svelte": "^3.1.4",
 		"@prisma/client": "^5.7.1",
 		"@svelte-put/clickoutside": "^3.0.1",

src/hooks.server.ts

@@ -0,0 +1,15 @@
+/** @type {import('@sveltejs/kit').HandleServerError} */
+export async function handle({ event, resolve }) {
+  const request = event.request;
+  const url = new URL(request.url);
+  if (url.pathname === "/") {
+    return new Response(null, {
+      status: 302,
+      headers: {
+        location: "/models"
+      }
+    });
+  }
+  return resolve(event);
+
+}

src/lib/utils/index.ts

@@ -39,7 +39,7 @@ export const MODELS_FILTER_OPTIONS = [
 export const SIDEBAR_MENUS = [ {
   icon: "uim:cube",
   label: "Models",
-  href: "/",
+  href: "/models",
 }, {
   icon: "solar:gallery-bold-duotone",
   label: "Gallery",

src/lib/utils/upload_image.ts

@@ -34,7 +34,6 @@ export const uploadImage = async (file: {
   try {
     // await client.send(putCommand)
     const data = await client.send(command);
-    console.log(data);
     // console.log('Successfully uploaded data to ' + bucketName + '/' + keyName)
   } catch (err: any) {
     console.error(err, err.stack)

src/routes/+page.server.ts

@@ -1,6 +1,6 @@
-import { redirect } from '@sveltejs/kit';
+// import { redirect } from '@sveltejs/kit';
 
 /** @type {import('./$types').PageLoad} */
 export async function load() {
-  throw redirect(302, "/models")
+  // throw redirect(302, "/models")
 }

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

@@ -70,20 +70,25 @@ export async function PATCH({ request } : RequestEvent) {
     }, { status: 401 });
   }
 
-  const hf_request = await fetch(`https://huggingface.co/api/models?limit=10000&filter=lora%2Cdiffusers&sort=likes7d`)
-  const hf_models = await hf_request.json();
+  const models = await prisma.model.findMany();
 
   let total_updates = 0;
 
-  for (const model of hf_models) {
+  for (const model of models) {
+    const hugging_face_request = await fetch(`https://huggingface.co/api/models/${model.id}`)
+    const hugging_face_model = await hugging_face_request.json()?.catch(() => {})
+
+    if (!hugging_face_model) {
+      continue;
+    }
     await prisma.model.update({
       where: {
         id: model.id
       },
       data: {
-        likes: model.likes,
-        downloads: model.downloads,
-        likes7d: model.likes7d,
+        likes: hugging_face_model.likes,
+        downloads: hugging_face_model.downloads,
+        likes7d: hugging_face_model.likes7d,
       }
     })
     .then(() => {

src/routes/gallery/[id]/+page.svelte

@@ -5,7 +5,6 @@
 
 
   onMount(() => {
-    console.log(data)
     galleryStore.set({
       gallery: data.gallery,
       open: true,

src/routes/models/+layout.svelte

@@ -2,6 +2,8 @@
 	import { browser } from "$app/environment";
 	import InfiniteScroll from "svelte-infinite-scroll";
 	import { page } from "$app/stores";
+	import { onMount } from "svelte";
+	// import { init } from "@huggingface/space-header";
 
 	import { goto } from "$app/navigation";
 	import Button from "$lib/components/Button.svelte";
@@ -10,13 +12,9 @@
 	import Radio from "$lib/components/fields/Radio.svelte";
 	import { MODELS_FILTER_OPTIONS } from "$lib/utils/index.js";
 	import GoTop from "$lib/components/GoTop.svelte";
-	import Dialog from "$lib/components/dialog/Dialog.svelte";
-	import SubmitModel from "$lib/components/models/Submit.svelte";
 	import Drawer from "$lib/components/models/drawer/Drawer.svelte";
-	import { onMount } from "svelte";
 	import type { ModelCard } from "$lib/type";
 	import { userStore } from "$lib/stores/use-user";
-	import Add from "$lib/components/community/reactions/Add.svelte";
 
 	let data: {
 		models: ModelCard[],
@@ -35,9 +33,7 @@
 
 	onMount(() => {
 		refetch(false);
-		if ($page.url.searchParams.get('model')) {
-			// getModel($page.url.searchParams.get('model') as string);
-		}
+		// init("enzostvs/lora-studio");
 	});
 
 	$: elementScroll = browser ? document?.getElementById('app') : undefined;

svelte.config.js

@@ -17,6 +17,11 @@ const config = {
 		}) : adapterNode({
 			out: 'build',
 		}),
+		prerender: {
+			entries: [
+				'*',
+			]
+		}
 	},
 };