Skip to content

Commit 5005353

Browse files
author
Mishig
authored
[Task Label & Icon] for Image-to-Video (#317)
This is how it looks <img width="407" alt="image" src="https://github.com/huggingface/huggingface.js/assets/11827707/428d78e9-0dc5-4a38-ac01-09936ed8e0e6">
1 parent 8d4520c commit 5005353

File tree

6 files changed

+15
-0
lines changed

6 files changed

+15
-0
lines changed

packages/hub/src/types/public.d.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ export type Task =
6969
| "text-to-image"
7070
| "image-to-text"
7171
| "image-to-image"
72+
| "image-to-video"
7273
| "unconditional-image-generation"
7374
| "video-classification"
7475
| "reinforcement-learning"

packages/tasks/src/const.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
1818
"image-segmentation": ["transformers", "transformers.js"],
1919
"image-to-image": [],
2020
"image-to-text": ["transformers.js"],
21+
"image-to-video": ["diffusers"],
2122
"video-classification": [],
2223
"mask-generation": ["transformers"],
2324
"multiple-choice": ["transformers"],

packages/tasks/src/pipelines.ts

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -438,6 +438,11 @@ export const PIPELINE_DATA = {
438438
modality: "cv",
439439
color: "indigo",
440440
},
441+
"image-to-video": {
442+
name: "Image-to-Video",
443+
modality: "multimodal",
444+
color: "indigo",
445+
},
441446
"unconditional-image-generation": {
442447
name: "Unconditional Image Generation",
443448
modality: "cv",

packages/tasks/src/tasksData.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
5454
"image-segmentation": getData("image-segmentation", imageSegmentation),
5555
"image-to-image": getData("image-to-image", imageToImage),
5656
"image-to-text": getData("image-to-text", imageToText),
57+
"image-to-video": undefined,
5758
"mask-generation": getData("mask-generation", placeholder),
5859
"multiple-choice": undefined,
5960
"object-detection": getData("object-detection", objectDetection),
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
<script lang="ts">
2+
export let classNames = "";
3+
</script>
4+
5+
<svg xmlns="http://www.w3.org/2000/svg" class={classNames} width="1em" height="1em" fill="none"><path fill="currentColor" fill-rule="evenodd" d="M1.43 1.2h6.63a.7.7 0 0 1 .7.7v4.44a.79.79 0 0 0-.34-.08h-.36V1.9H1.43v3.55l.9-.9a.7.7 0 0 1 .99 0l1.74 1.74a.79.79 0 0 0-.51.47L2.82 5.04l-1.4 1.4v2.09H4.5v.7H1.43a.7.7 0 0 1-.7-.7V1.9a.7.7 0 0 1 .7-.7ZM6.8 5.94l.32.32H5.5l.32-.32a.7.7 0 0 1 .99 0ZM6.2 4.52a1.05 1.05 0 1 1-1.17-1.75A1.05 1.05 0 0 1 6.2 4.52Zm-.4-1.16a.35.35 0 1 0-.38.58.35.35 0 0 0 .39-.58Zm3.9 5.16 1.57-1.26v3.14L9.7 9.15V10a.79.79 0 0 1-.79.79H5.77a.79.79 0 0 1-.79-.79V7.65a.79.79 0 0 1 .79-.78H8.9a.79.79 0 0 1 .79.78v.87Zm-3.93-.87v2.36H8.9V7.65H5.77Z" clip-rule="evenodd"/></svg>

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import IconTextGeneration from "../Icons/IconTextGeneration.svelte";
2424
import IconTextToImage from "../Icons/IconTextToImage.svelte";
2525
import IconImageToText from "../Icons/IconImageToText.svelte";
26+
import IconImageToVideo from "../Icons/IconImageToVideo.svelte";
2627
import IconTextToSpeech from "../Icons/IconTextToSpeech.svelte";
2728
import IconTextToVideo from "../Icons/IconTextToVideo.svelte";
2829
import IconTokenClassification from "../Icons/IconTokenClassification.svelte";
@@ -74,6 +75,7 @@
7475
"text-to-image": IconTextToImage,
7576
"image-to-text": IconImageToText,
7677
"image-to-image": IconImageToImage,
78+
"image-to-video": IconImageToVideo,
7779
"unconditional-image-generation": IconUnconditionalImageGeneration,
7880
"reinforcement-learning": IconReinforcementLearning,
7981
robotics: IconRobotics,

0 commit comments

Comments
 (0)