Skip to content

Commit 8b50438

Browse files
Add document-retrieval to Hub as a task (#1097)
--------- Co-authored-by: vb <[email protected]>
1 parent 499c9a7 commit 8b50438

File tree

4 files changed

+11
-0
lines changed

4 files changed

+11
-0
lines changed

packages/tasks/src/pipelines.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -676,6 +676,12 @@ export const PIPELINE_DATA = {
676676
color: "red",
677677
hideInDatasets: true,
678678
},
679+
"visual-document-retrieval": {
680+
name: "Visual Document Retrieval",
681+
modality: "multimodal",
682+
color: "yellow",
683+
hideInDatasets: true,
684+
},
679685
"any-to-any": {
680686
name: "Any-to-Any",
681687
modality: "multimodal",

packages/tasks/src/tasks/index.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
174174
"text-to-3d": ["diffusers"],
175175
"image-to-3d": ["diffusers"],
176176
"any-to-any": ["transformers"],
177+
"visual-document-retrieval": ["transformers"],
177178
};
178179

179180
/**
@@ -202,6 +203,7 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
202203
"automatic-speech-recognition": getData("automatic-speech-recognition", automaticSpeechRecognition),
203204
"depth-estimation": getData("depth-estimation", depthEstimation),
204205
"document-question-answering": getData("document-question-answering", documentQuestionAnswering),
206+
"visual-document-retrieval": getData("visual-document-retrieval", placeholder),
205207
"feature-extraction": getData("feature-extraction", featureExtraction),
206208
"fill-mask": getData("fill-mask", fillMask),
207209
"graph-ml": undefined,
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
<svg xmlns="http://www.w3.org/2000/svg" width="1em" height="1em" viewBox="0 0 32 32" {...$$props}><path fill="currentColor" d="m27.4 14.7l-6.1-6.1C21 8.2 20.5 8 20 8h-8c-1.1 0-2 .9-2 2v18c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V16.1c0-.5-.2-1-.6-1.4M20 10l5.9 6H20zm-8 18V10h6v6c0 1.1.9 2 2 2h6v10z"/><path fill="currentColor" d="M6 18H4V4c0-1.1.9-2 2-2h14v2H6z"/></svg>

packages/widgets/src/lib/components/PipelineIcon/PipelineIcon.svelte

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
import IconVideoTextToText from "../Icons/IconVideoTextToText.svelte";
4646
import IconAnyToAny from "../Icons/IconAnyToAny.svelte";
4747
import IconKeypointDetection from "../Icons/IconKeypointDetection.svelte";
48+
import IconVisualDocumentRetrieval from "../IconsVisualDocumentRetrieval.svelte";
4849
import type { WidgetType } from "@huggingface/tasks";
4950
5051
export let classNames = "";
@@ -101,6 +102,7 @@
101102
"video-text-to-text": IconVideoTextToText,
102103
"any-to-any": IconAnyToAny,
103104
"keypoint-detection": IconKeypointDetection,
105+
"visual-document-retrieval": IconVisualDocumentRetrieval,
104106
};
105107
106108
$: iconComponent =

0 commit comments

Comments
 (0)