Skip to content

Commit bb8e9c3

Browse files
authored
Merge branch 'main' into asr
2 parents d170516 + 27b995d commit bb8e9c3

File tree

3 files changed

+26
-2
lines changed

3 files changed

+26
-2
lines changed

packages/tasks/package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
{
22
"name": "@huggingface/tasks",
33
"packageManager": "[email protected]",
4-
"version": "0.12.30",
4+
"version": "0.13.0",
55
"description": "List of ML tasks for huggingface.co/tasks",
66
"repository": "https://github.com/huggingface/huggingface.js.git",
77
"publishConfig": {

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -363,6 +363,23 @@ export const gliner = (model: ModelData): string[] => [
363363
model = GLiNER.from_pretrained("${model.id}")`,
364364
];
365365

366+
export const htrflow = (model: ModelData): string[] => [
367+
`# CLI usage
368+
# see docs: https://ai-riksarkivet.github.io/htrflow/latest/getting_started/quick_start.html
369+
htrflow pipeline <path/to/pipeline.yaml> <path/to/image>`,
370+
`# Python usage
371+
from htrflow.pipeline.pipeline import Pipeline
372+
from htrflow.pipeline.steps import Task
373+
from htrflow.models.framework.model import ModelClass
374+
375+
pipeline = Pipeline(
376+
[
377+
Task(
378+
ModelClass, {"model": "${model.id}"}, {}
379+
),
380+
])`,
381+
];
382+
366383
export const keras = (model: ModelData): string[] => [
367384
`# Available backend options are: "jax", "torch", "tensorflow".
368385
import os

packages/tasks/src/model-libraries.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,6 +317,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
317317
docsUrl: "https://hezarai.github.io/hezar",
318318
countDownloads: `path:"model_config.yaml" OR path:"embedding/embedding_config.yaml"`,
319319
},
320+
htrflow: {
321+
prettyLabel: "HTRflow",
322+
repoName: "HTRflow",
323+
repoUrl: "https://github.com/AI-Riksarkivet/htrflow",
324+
docsUrl: "https://ai-riksarkivet.github.io/htrflow",
325+
snippets: snippets.htrflow,
326+
},
320327
"hunyuan-dit": {
321328
prettyLabel: "HunyuanDiT",
322329
repoName: "HunyuanDiT",
@@ -473,7 +480,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
473480
prettyLabel: "open-oasis",
474481
repoName: "open-oasis",
475482
repoUrl: "https://github.com/etched-ai/open-oasis",
476-
countDownloads: `path:"oasis500m.pt"`,
483+
countDownloads: `path:"oasis500m.safetensors"`,
477484
},
478485
open_clip: {
479486
prettyLabel: "OpenCLIP",

0 commit comments

Comments
 (0)