Upload 103 files

ui/src/app/api/hf-hub/route.ts

@@ -1,8 +1,7 @@
 import { NextRequest, NextResponse } from 'next/server';
 import { whoAmI, createRepo, uploadFiles, datasetInfo } from '@huggingface/hub';
-import { readdir, stat } from 'fs/promises';
+import { readdir, stat, readFile } from 'fs/promises';
 import path from 'path';
-import { pathToFileURL } from 'url';
 
 export async function POST(request: NextRequest) {
   try {
@@ -73,8 +72,9 @@ export async function POST(request: NextRequest) {
             filesToUpload.push({ path: normalizedRepoPath, content });
           };
 
-          const addUploadFile = (absolutePath: string, repoFilePath: string) => {
-            addUploadContent(repoFilePath, pathToFileURL(absolutePath));
+          const addUploadFile = async (absolutePath: string, repoFilePath: string) => {
+            const buffer = await readFile(absolutePath);
+            addUploadContent(repoFilePath, buffer);
           };
 
           const walkDirectory = async (basePath: string, repoPrefix: string) => {
@@ -86,7 +86,7 @@ export async function POST(request: NextRequest) {
                 await walkDirectory(entryPath, nextPrefix);
               } else if (entry.isFile()) {
                 const repoFilePath = repoPrefix ? `${repoPrefix}/${entry.name}` : entry.name;
-                addUploadFile(entryPath, repoFilePath);
+                await addUploadFile(entryPath, repoFilePath);
               }
             }
           };
@@ -111,7 +111,7 @@ export async function POST(request: NextRequest) {
               } else if (!path.posix.extname(destination)) {
                 destination = `${destination}/${path.basename(resolvedPath)}`;
               }
-              addUploadFile(resolvedPath, destination);
+              await addUploadFile(resolvedPath, destination);
             } else {
               throw new Error(`Unsupported artifact type for path: ${localPath}`);
             }

ui/src/app/api/hf-jobs/route.ts

@@ -615,9 +615,9 @@ def generate_model_card_readme(repo_id: str, config: dict, model_name: str, cura
             })
             prompt_bullets.append(f"- {prompt_text}")
 
-        gallery_section = "<Gallery />\n\n"
+        gallery_section = ""
         if prompt_bullets:
-            gallery_section += "### Prompts\n\n" + "\n".join(prompt_bullets) + "\n\n"
+            gallery_section = "<Gallery />\n\n" + "### Prompts\n\n" + "\n".join(prompt_bullets) + "\n\n"
 
         # Determine torch dtype based on model
         dtype = "torch.bfloat16" if "flux" in arch.lower() else "torch.float16"

ui/src/paths.ts

@@ -1,5 +1,8 @@
 import path from 'path';
-export const TOOLKIT_ROOT = path.resolve('@', '..', '..');
-export const defaultTrainFolder = path.join(TOOLKIT_ROOT, 'output');
-export const defaultDatasetsFolder = path.join(TOOLKIT_ROOT, 'datasets');
-export const defaultDataRoot = path.join(TOOLKIT_ROOT, 'data');
+
+const WORKSPACE_ROOT = path.resolve(process.cwd(), '..');
+
+export const TOOLKIT_ROOT = WORKSPACE_ROOT;
+export const defaultTrainFolder = path.join(WORKSPACE_ROOT, 'output');
+export const defaultDatasetsFolder = path.join(WORKSPACE_ROOT, 'datasets');
+export const defaultDataRoot = path.join(WORKSPACE_ROOT, 'data');