webml-kit 0.1.0 → 0.2.0

package/README.md

@@ -1,3 +1,7 @@
+<p align="center">
+  <img src="logo.png" alt="webml-kit" width="200" />
+</p>
+
 # webml-kit
 
 > Framework-agnostic utilities for loading and running ML models in the browser via WebGPU/WASM.
@@ -9,7 +13,7 @@ This library does that part for you. It wraps [`@huggingface/transformers`](http
 ## Install
 
 ```bash
-npm install webml-kit @huggingface/transformers
+npm install webml-kit
 ```
 
 ## Quick start
@@ -58,6 +62,26 @@ const { ok, reason } = await canRun('4GB');    // human-readable
 const same = await canRun(4_000_000_000);       // raw bytes — same result
 ```
 
+### Model discovery
+
+Search the Hugging Face Hub for models that work with webml-kit. Only returns models tagged with `transformers.js`, so everything listed runs out of the box:
+
+```ts
+import { searchModels, listModelsForTask, trendingModels, listWebGPUModels } from 'webml-kit';
+
+// Search by keyword
+const results = await searchModels({ query: 'whisper', task: 'automatic-speech-recognition' });
+
+// Top models for a task
+const textModels = await listModelsForTask('text-generation', 5);
+
+// What's trending right now
+const hot = await trendingModels(5);
+
+// Only from known WebGPU-friendly orgs (onnx-community, Xenova)
+const webgpu = await listWebGPUModels({ task: 'text-generation' });
+```
+
 ### Cache visibility
 
 The worst UX in browser ML is showing "downloading 2GB..." to someone who already has the model. Now you can check:
@@ -157,6 +181,11 @@ These work without a ModelClient — useful for pre-flight checks:
 | `getCacheSize()` | Total bytes used by cached models |
 | `parseSize(input)` | Convert '4GB' / '512MB' to bytes |
 | `formatSize(bytes)` | Convert bytes to '4.0 GB' / '512.0 MB' |
+| `searchModels(options?)` | Search HF Hub for transformers.js-compatible models |
+| `listModelsForTask(task)` | Top models for a specific pipeline task |
+| `trendingModels(limit?)` | What's trending on HF Hub right now |
+| `listWebGPUModels(options?)` | Models from known WebGPU-friendly orgs |
+| `getModelInfo(modelId)` | Detailed info about a specific model |
 
 ## Supported tasks
 

package/dist/hub.d.ts

@@ -0,0 +1,97 @@
+/**
+ * Search and list WebGPU/ONNX-compatible models from the Hugging Face Hub.
+ *
+ * Uses the public HF Hub API — no auth token needed for public models.
+ */
+import type { PipelineTask } from './types.js';
+export interface HubModel {
+    /** Model ID (e.g. 'onnx-community/Bonsai-1.7B-ONNX') */
+    modelId: string;
+    /** Pipeline task (e.g. 'text-generation') */
+    task: PipelineTask | string;
+    /** Author or organization */
+    author: string;
+    /** Number of downloads in the last month */
+    downloads: number;
+    /** Number of likes */
+    likes: number;
+    /** Last modified date */
+    lastModified: string;
+    /** Tags on the model (e.g. ['onnx', 'webgpu']) */
+    tags: string[];
+    /** Whether it's likely WebGPU-compatible (has ONNX weights) */
+    webgpuCompatible: boolean;
+}
+export interface SearchOptions {
+    /** Filter by pipeline task */
+    task?: PipelineTask | string;
+    /** Free-text search query */
+    query?: string;
+    /** Sort by: 'downloads' | 'likes' | 'modified' | 'trending' */
+    sort?: 'downloads' | 'likes' | 'modified' | 'trending';
+    /** Sort direction */
+    direction?: 'asc' | 'desc';
+    /** Max results to return (default: 20, max: 100) */
+    limit?: number;
+    /** Filter by author/org (e.g. 'onnx-community', 'Xenova') */
+    author?: string;
+}
+/** Known orgs that publish ONNX/WebGPU-ready models */
+export declare const WEBGPU_ORGS: readonly ["onnx-community", "Xenova", "webml-community"];
+/**
+ * Search for models compatible with transformers.js (and thus WebGPU/WASM).
+ *
+ * ```ts
+ * import { searchModels } from 'webml-kit';
+ *
+ * const models = await searchModels({ task: 'text-generation', limit: 5 });
+ * for (const m of models) {
+ *   console.log(`${m.modelId} — ${m.downloads} downloads`);
+ * }
+ * ```
+ */
+export declare function searchModels(options?: SearchOptions): Promise<HubModel[]>;
+/**
+ * List popular models for a specific task, sorted by downloads.
+ *
+ * ```ts
+ * import { listModelsForTask } from 'webml-kit';
+ *
+ * const models = await listModelsForTask('automatic-speech-recognition');
+ * // [{ modelId: 'onnx-community/whisper-tiny.en', downloads: 12345, ... }]
+ * ```
+ */
+export declare function listModelsForTask(task: PipelineTask | string, limit?: number): Promise<HubModel[]>;
+/**
+ * List trending transformers.js models right now.
+ *
+ * ```ts
+ * import { trendingModels } from 'webml-kit';
+ *
+ * const hot = await trendingModels(5);
+ * ```
+ */
+export declare function trendingModels(limit?: number): Promise<HubModel[]>;
+/**
+ * List all models from known WebGPU-friendly orgs
+ * (onnx-community, Xenova, webml-community).
+ *
+ * ```ts
+ * import { listWebGPUModels } from 'webml-kit';
+ *
+ * const models = await listWebGPUModels({ task: 'text-generation' });
+ * ```
+ */
+export declare function listWebGPUModels(options?: Omit<SearchOptions, 'author'>): Promise<HubModel[]>;
+/**
+ * Get detailed info about a specific model.
+ *
+ * ```ts
+ * import { getModelInfo } from 'webml-kit';
+ *
+ * const info = await getModelInfo('onnx-community/Bonsai-1.7B-ONNX');
+ * console.log(info.task, info.downloads, info.webgpuCompatible);
+ * ```
+ */
+export declare function getModelInfo(modelId: string): Promise<HubModel>;
+//# sourceMappingURL=hub.d.ts.map

package/dist/hub.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hub.d.ts","sourceRoot":"","sources":["../src/hub.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAI/C,MAAM,WAAW,QAAQ;IACvB,wDAAwD;IACxD,OAAO,EAAE,MAAM,CAAC;IAChB,6CAA6C;IAC7C,IAAI,EAAE,YAAY,GAAG,MAAM,CAAC;IAC5B,6BAA6B;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,4CAA4C;IAC5C,SAAS,EAAE,MAAM,CAAC;IAClB,sBAAsB;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,yBAAyB;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,kDAAkD;IAClD,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,+DAA+D;IAC/D,gBAAgB,EAAE,OAAO,CAAC;CAC3B;AAED,MAAM,WAAW,aAAa;IAC5B,8BAA8B;IAC9B,IAAI,CAAC,EAAE,YAAY,GAAG,MAAM,CAAC;IAC7B,6BAA6B;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,+DAA+D;IAC/D,IAAI,CAAC,EAAE,WAAW,GAAG,OAAO,GAAG,UAAU,GAAG,UAAU,CAAC;IACvD,qBAAqB;IACrB,SAAS,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;IAC3B,oDAAoD;IACpD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,6DAA6D;IAC7D,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAMD,uDAAuD;AACvD,eAAO,MAAM,WAAW,0DAId,CAAC;AAIX;;;;;;;;;;;GAWG;AACH,wBAAsB,YAAY,CAAC,OAAO,GAAE,aAAkB,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CA8BnF;AAED;;;;;;;;;GASG;AACH,wBAAsB,iBAAiB,CACrC,IAAI,EAAE,YAAY,GAAG,MAAM,EAC3B,KAAK,SAAK,GACT,OAAO,CAAC,QAAQ,EAAE,CAAC,CAErB;AAED;;;;;;;;GAQG;AACH,wBAAsB,cAAc,CAAC,KAAK,SAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAEpE;AAED;;;;;;;;;GASG;AACH,wBAAsB,gBAAgB,CACpC,OAAO,GAAE,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAM,GAC1C,OAAO,CAAC,QAAQ,EAAE,CAAC,CAiBrB;AAED;;;;;;;;;GASG;AACH,wBAAsB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,CAQrE"}

package/dist/hub.js

@@ -0,0 +1,142 @@
+/**
+ * Search and list WebGPU/ONNX-compatible models from the Hugging Face Hub.
+ *
+ * Uses the public HF Hub API — no auth token needed for public models.
+ */
+// ─── Constants ───
+const HF_API = 'https://huggingface.co/api';
+/** Known orgs that publish ONNX/WebGPU-ready models */
+export const WEBGPU_ORGS = [
+    'onnx-community',
+    'Xenova',
+    'webml-community',
+];
+// ─── Core functions ───
+/**
+ * Search for models compatible with transformers.js (and thus WebGPU/WASM).
+ *
+ * ```ts
+ * import { searchModels } from 'webml-kit';
+ *
+ * const models = await searchModels({ task: 'text-generation', limit: 5 });
+ * for (const m of models) {
+ *   console.log(`${m.modelId} — ${m.downloads} downloads`);
+ * }
+ * ```
+ */
+export async function searchModels(options = {}) {
+    const params = new URLSearchParams();
+    // Only show models that work with transformers.js
+    params.set('library', 'transformers.js');
+    if (options.task)
+        params.set('pipeline_tag', options.task);
+    if (options.query)
+        params.set('search', options.query);
+    if (options.author)
+        params.set('author', options.author);
+    if (options.sort === 'trending') {
+        params.set('sort', 'trending');
+    }
+    else if (options.sort) {
+        params.set('sort', options.sort === 'modified' ? 'lastModified' : options.sort);
+        params.set('direction', options.direction === 'asc' ? '1' : '-1');
+    }
+    else {
+        params.set('sort', 'downloads');
+        params.set('direction', '-1');
+    }
+    params.set('limit', String(Math.min(options.limit ?? 20, 100)));
+    const url = `${HF_API}/models?${params.toString()}`;
+    const response = await fetch(url);
+    if (!response.ok) {
+        throw new Error(`HF Hub API error: ${response.status} ${response.statusText}`);
+    }
+    const data = await response.json();
+    return data.map(normalizeModel);
+}
+/**
+ * List popular models for a specific task, sorted by downloads.
+ *
+ * ```ts
+ * import { listModelsForTask } from 'webml-kit';
+ *
+ * const models = await listModelsForTask('automatic-speech-recognition');
+ * // [{ modelId: 'onnx-community/whisper-tiny.en', downloads: 12345, ... }]
+ * ```
+ */
+export async function listModelsForTask(task, limit = 10) {
+    return searchModels({ task, sort: 'downloads', limit });
+}
+/**
+ * List trending transformers.js models right now.
+ *
+ * ```ts
+ * import { trendingModels } from 'webml-kit';
+ *
+ * const hot = await trendingModels(5);
+ * ```
+ */
+export async function trendingModels(limit = 10) {
+    return searchModels({ sort: 'trending', limit });
+}
+/**
+ * List all models from known WebGPU-friendly orgs
+ * (onnx-community, Xenova, webml-community).
+ *
+ * ```ts
+ * import { listWebGPUModels } from 'webml-kit';
+ *
+ * const models = await listWebGPUModels({ task: 'text-generation' });
+ * ```
+ */
+export async function listWebGPUModels(options = {}) {
+    const results = [];
+    for (const org of WEBGPU_ORGS) {
+        const models = await searchModels({ ...options, author: org });
+        results.push(...models);
+    }
+    // Deduplicate by modelId and sort by downloads
+    const seen = new Set();
+    return results
+        .filter((m) => {
+        if (seen.has(m.modelId))
+            return false;
+        seen.add(m.modelId);
+        return true;
+    })
+        .sort((a, b) => b.downloads - a.downloads);
+}
+/**
+ * Get detailed info about a specific model.
+ *
+ * ```ts
+ * import { getModelInfo } from 'webml-kit';
+ *
+ * const info = await getModelInfo('onnx-community/Bonsai-1.7B-ONNX');
+ * console.log(info.task, info.downloads, info.webgpuCompatible);
+ * ```
+ */
+export async function getModelInfo(modelId) {
+    const response = await fetch(`${HF_API}/models/${modelId}`);
+    if (!response.ok) {
+        throw new Error(`Model not found: ${modelId} (${response.status})`);
+    }
+    const data = await response.json();
+    return normalizeModel(data);
+}
+function normalizeModel(raw) {
+    const id = raw.modelId ?? raw.id ?? 'unknown';
+    const tags = raw.tags ?? [];
+    const author = raw.author ?? id.split('/')[0] ?? 'unknown';
+    return {
+        modelId: id,
+        task: (raw.pipeline_tag ?? 'unknown'),
+        author,
+        downloads: raw.downloads ?? 0,
+        likes: raw.likes ?? 0,
+        lastModified: raw.lastModified ?? '',
+        tags,
+        webgpuCompatible: tags.includes('onnx') || WEBGPU_ORGS.some((org) => id.startsWith(org)),
+    };
+}
+//# sourceMappingURL=hub.js.map

package/dist/hub.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hub.js","sourceRoot":"","sources":["../src/hub.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAwCH,oBAAoB;AAEpB,MAAM,MAAM,GAAG,4BAA4B,CAAC;AAE5C,uDAAuD;AACvD,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,gBAAgB;IAChB,QAAQ;IACR,iBAAiB;CACT,CAAC;AAEX,yBAAyB;AAEzB;;;;;;;;;;;GAWG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,UAAyB,EAAE;IAC5D,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;IAErC,kDAAkD;IAClD,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;IAEzC,IAAI,OAAO,CAAC,IAAI;QAAE,MAAM,CAAC,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3D,IAAI,OAAO,CAAC,KAAK;QAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IACvD,IAAI,OAAO,CAAC,MAAM;QAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IACzD,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAChC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACjC,CAAC;SAAM,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QACxB,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAChF,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAChC,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IAEhE,MAAM,GAAG,GAAG,GAAG,MAAM,WAAW,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;IAEpD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;IAClC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;IACjF,CAAC;IAED,MAAM,IAAI,GAAkB,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAClD,OAAO,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;AAClC,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAA2B,EAC3B,KAAK,GAAG,EAAE;IAEV,OAAO,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;AAC1D,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,KAAK,GAAG,EAAE;IAC7C,OAAO,YAAY,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;AACnD,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,UAAyC,EAAE;IAE3C,MAAM,OAAO,GAAe,EAAE,CAAC;IAE/B,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/D,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;IAC1B,CAAC;IAED,+CAA+C;IAC/C,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,OAAO,OAAO;SACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;QACZ,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;YAAE,OAAO,KAAK,CAAC;QACtC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;SACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;AAC/C,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,OAAe;IAChD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,MAAM,WAAW,OAAO,EAAE,CAAC,CAAC;IAC5D,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,oBAAoB,OAAO,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,IAAI,GAAgB,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAChD,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAeD,SAAS,cAAc,CAAC,GAAgB;IACtC,MAAM,EAAE,GAAG,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,EAAE,IAAI,SAAS,CAAC;IAC9C,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;IAE3D,OAAO;QACL,OAAO,EAAE,EAAE;QACX,IAAI,EAAE,CAAC,GAAG,CAAC,YAAY,IAAI,SAAS,CAA0B;QAC9D,MAAM;QACN,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,CAAC;QAC7B,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,CAAC;QACrB,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,EAAE;QACpC,IAAI;QACJ,gBAAgB,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;KACzF,CAAC;AACJ,CAAC"}

package/dist/index.d.ts

@@ -47,5 +47,7 @@ export { GPURecovery } from './gpu-recovery.js';
 export type { RecoveryState, GPURecoveryEvents } from './gpu-recovery.js';
 export { PIPELINE_REGISTRY, getPipelineDefaults, supportsStreaming, } from './pipelines/index.js';
 export type { PipelineDefaults } from './pipelines/index.js';
+export { searchModels, listModelsForTask, trendingModels, listWebGPUModels, getModelInfo, WEBGPU_ORGS, } from './hub.js';
+export type { HubModel, SearchOptions } from './hub.js';
 export type { DeviceBackend, DeviceInfo, GPUInfo, QuantizationType, PipelineTask, ModelConfig, ProgressEvent, ProgressCallback, TokenEvent, ChatMessage, TextGenerationOptions, TextGenerationResult, ClassificationResult, DetectionResult, TranscriptionResult, EmbeddingResult, CaptionResult, TextResult, CacheBackend, CachedModel, } from './types.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuCG;AAGH,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGtE,OAAO,EACL,YAAY,EACZ,WAAW,EACX,SAAS,EACT,aAAa,EACb,UAAU,EACV,cAAc,EACd,MAAM,EACN,SAAS,EACT,UAAU,GACX,MAAM,aAAa,CAAC;AAGrB,OAAO,EACL,eAAe,EACf,QAAQ,EACR,YAAY,EACZ,gBAAgB,EAChB,UAAU,GACX,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAG5D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,YAAY,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAG1E,OAAO,EACL,iBAAiB,EACjB,mBAAmB,EACnB,iBAAiB,GAClB,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAG7D,YAAY,EACV,aAAa,EACb,UAAU,EACV,OAAO,EACP,gBAAgB,EAChB,YAAY,EACZ,WAAW,EACX,aAAa,EACb,gBAAgB,EAChB,UAAU,EACV,WAAW,EACX,qBAAqB,EACrB,oBAAoB,EACpB,oBAAoB,EACpB,eAAe,EACf,mBAAmB,EACnB,eAAe,EACf,aAAa,EACb,UAAU,EACV,YAAY,EACZ,WAAW,GACZ,MAAM,YAAY,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuCG;AAGH,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGtE,OAAO,EACL,YAAY,EACZ,WAAW,EACX,SAAS,EACT,aAAa,EACb,UAAU,EACV,cAAc,EACd,MAAM,EACN,SAAS,EACT,UAAU,GACX,MAAM,aAAa,CAAC;AAGrB,OAAO,EACL,eAAe,EACf,QAAQ,EACR,YAAY,EACZ,gBAAgB,EAChB,UAAU,GACX,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAG5D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,YAAY,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAG1E,OAAO,EACL,iBAAiB,EACjB,mBAAmB,EACnB,iBAAiB,GAClB,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAG7D,OAAO,EACL,YAAY,EACZ,iBAAiB,EACjB,cAAc,EACd,gBAAgB,EAChB,YAAY,EACZ,WAAW,GACZ,MAAM,UAAU,CAAC;AAClB,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAGxD,YAAY,EACV,aAAa,EACb,UAAU,EACV,OAAO,EACP,gBAAgB,EAChB,YAAY,EACZ,WAAW,EACX,aAAa,EACb,gBAAgB,EAChB,UAAU,EACV,WAAW,EACX,qBAAqB,EACrB,oBAAoB,EACpB,oBAAoB,EACpB,eAAe,EACf,mBAAmB,EACnB,eAAe,EACf,aAAa,EACb,UAAU,EACV,YAAY,EACZ,WAAW,GACZ,MAAM,YAAY,CAAC"}

package/dist/index.js

@@ -50,4 +50,6 @@ export { TokenStream, collectStream } from './streaming.js';
 export { GPURecovery } from './gpu-recovery.js';
 // ─── Pipelines ───
 export { PIPELINE_REGISTRY, getPipelineDefaults, supportsStreaming, } from './pipelines/index.js';
+// ─── Hub Search ───
+export { searchModels, listModelsForTask, trendingModels, listWebGPUModels, getModelInfo, WEBGPU_ORGS, } from './hub.js';
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuCG;AAEH,eAAe;AACf,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhD,2BAA2B;AAC3B,OAAO,EACL,YAAY,EACZ,WAAW,EACX,SAAS,EACT,aAAa,EACb,UAAU,EACV,cAAc,EACd,MAAM,EACN,SAAS,EACT,UAAU,GACX,MAAM,aAAa,CAAC;AAErB,gBAAgB;AAChB,OAAO,EACL,eAAe,EACf,QAAQ,EACR,YAAY,EACZ,gBAAgB,EAChB,UAAU,GACX,MAAM,YAAY,CAAC;AAEpB,oBAAoB;AACpB,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE5D,uBAAuB;AACvB,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhD,oBAAoB;AACpB,OAAO,EACL,iBAAiB,EACjB,mBAAmB,EACnB,iBAAiB,GAClB,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuCG;AAEH,eAAe;AACf,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhD,2BAA2B;AAC3B,OAAO,EACL,YAAY,EACZ,WAAW,EACX,SAAS,EACT,aAAa,EACb,UAAU,EACV,cAAc,EACd,MAAM,EACN,SAAS,EACT,UAAU,GACX,MAAM,aAAa,CAAC;AAErB,gBAAgB;AAChB,OAAO,EACL,eAAe,EACf,QAAQ,EACR,YAAY,EACZ,gBAAgB,EAChB,UAAU,GACX,MAAM,YAAY,CAAC;AAEpB,oBAAoB;AACpB,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE5D,uBAAuB;AACvB,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhD,oBAAoB;AACpB,OAAO,EACL,iBAAiB,EACjB,mBAAmB,EACnB,iBAAiB,GAClB,MAAM,sBAAsB,CAAC;AAG9B,qBAAqB;AACrB,OAAO,EACL,YAAY,EACZ,iBAAiB,EACjB,cAAc,EACd,gBAAgB,EAChB,YAAY,EACZ,WAAW,GACZ,MAAM,UAAU,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "webml-kit",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Framework-agnostic utilities for loading and running ML models in the browser via WebGPU/WASM",
   "type": "module",
   "main": "./dist/index.js",
@@ -17,7 +17,8 @@
   },
   "files": [
     "dist",
-    "README.md"
+    "README.md",
+    "logo.png"
   ],
   "scripts": {
     "build": "tsc",
@@ -44,11 +45,10 @@
     "type": "git",
     "url": "https://github.com/hemanth/webml-kit.git"
   },
-  "peerDependencies": {
-    "@huggingface/transformers": ">=4.0.0"
+  "dependencies": {
+    "@huggingface/transformers": "^4.1.0"
   },
   "devDependencies": {
-    "@huggingface/transformers": "^4.1.0",
     "@webgpu/types": "^0.1.69",
     "typescript": "^5.7.0",
     "vitest": "^4.1.4"