@@ -43,10 +43,6 @@ export interface PipelineData {
4343 name : string ;
4444 subtasks ?: SubTask [ ] ;
4545 modality : Modality ;
46- /**
47- * color for the tag icon.
48- */
49- color : "blue" | "green" | "indigo" | "orange" | "red" | "yellow" ;
5046 /**
5147 * whether to hide in /models filters
5248 */
@@ -143,7 +139,6 @@ export const PIPELINE_DATA = {
143139 } ,
144140 ] ,
145141 modality : "nlp" ,
146- color : "orange" ,
147142 } ,
148143 "token-classification" : {
149144 name : "Token Classification" ,
@@ -174,12 +169,10 @@ export const PIPELINE_DATA = {
174169 } ,
175170 ] ,
176171 modality : "nlp" ,
177- color : "blue" ,
178172 } ,
179173 "table-question-answering" : {
180174 name : "Table Question Answering" ,
181175 modality : "nlp" ,
182- color : "green" ,
183176 } ,
184177 "question-answering" : {
185178 name : "Question Answering" ,
@@ -198,17 +191,14 @@ export const PIPELINE_DATA = {
198191 } ,
199192 ] ,
200193 modality : "nlp" ,
201- color : "blue" ,
202194 } ,
203195 "zero-shot-classification" : {
204196 name : "Zero-Shot Classification" ,
205197 modality : "nlp" ,
206- color : "yellow" ,
207198 } ,
208199 translation : {
209200 name : "Translation" ,
210201 modality : "nlp" ,
211- color : "green" ,
212202 } ,
213203 summarization : {
214204 name : "Summarization" ,
@@ -223,12 +213,10 @@ export const PIPELINE_DATA = {
223213 } ,
224214 ] ,
225215 modality : "nlp" ,
226- color : "indigo" ,
227216 } ,
228217 "feature-extraction" : {
229218 name : "Feature Extraction" ,
230219 modality : "nlp" ,
231- color : "red" ,
232220 } ,
233221 "text-generation" : {
234222 name : "Text Generation" ,
@@ -251,7 +239,6 @@ export const PIPELINE_DATA = {
251239 } ,
252240 ] ,
253241 modality : "nlp" ,
254- color : "indigo" ,
255242 } ,
256243 "text2text-generation" : {
257244 name : "Text2Text Generation" ,
@@ -286,7 +273,6 @@ export const PIPELINE_DATA = {
286273 } ,
287274 ] ,
288275 modality : "nlp" ,
289- color : "indigo" ,
290276 } ,
291277 "fill-mask" : {
292278 name : "Fill-Mask" ,
@@ -301,32 +287,26 @@ export const PIPELINE_DATA = {
301287 } ,
302288 ] ,
303289 modality : "nlp" ,
304- color : "red" ,
305290 } ,
306291 "sentence-similarity" : {
307292 name : "Sentence Similarity" ,
308293 modality : "nlp" ,
309- color : "yellow" ,
310294 } ,
311295 "text-to-speech" : {
312296 name : "Text-to-Speech" ,
313297 modality : "audio" ,
314- color : "yellow" ,
315298 } ,
316299 "text-to-audio" : {
317300 name : "Text-to-Audio" ,
318301 modality : "audio" ,
319- color : "yellow" ,
320302 } ,
321303 "automatic-speech-recognition" : {
322304 name : "Automatic Speech Recognition" ,
323305 modality : "audio" ,
324- color : "yellow" ,
325306 } ,
326307 "audio-to-audio" : {
327308 name : "Audio-to-Audio" ,
328309 modality : "audio" ,
329- color : "blue" ,
330310 } ,
331311 "audio-classification" : {
332312 name : "Audio Classification" ,
@@ -353,23 +333,19 @@ export const PIPELINE_DATA = {
353333 } ,
354334 ] ,
355335 modality : "audio" ,
356- color : "green" ,
357336 } ,
358337 "audio-text-to-text" : {
359338 name : "Audio-Text-to-Text" ,
360339 modality : "multimodal" ,
361- color : "red" ,
362340 hideInDatasets : true ,
363341 } ,
364342 "voice-activity-detection" : {
365343 name : "Voice Activity Detection" ,
366344 modality : "audio" ,
367- color : "red" ,
368345 } ,
369346 "depth-estimation" : {
370347 name : "Depth Estimation" ,
371348 modality : "cv" ,
372- color : "yellow" ,
373349 } ,
374350 "image-classification" : {
375351 name : "Image Classification" ,
@@ -384,7 +360,6 @@ export const PIPELINE_DATA = {
384360 } ,
385361 ] ,
386362 modality : "cv" ,
387- color : "blue" ,
388363 } ,
389364 "object-detection" : {
390365 name : "Object Detection" ,
@@ -399,7 +374,6 @@ export const PIPELINE_DATA = {
399374 } ,
400375 ] ,
401376 modality : "cv" ,
402- color : "yellow" ,
403377 } ,
404378 "image-segmentation" : {
405379 name : "Image Segmentation" ,
@@ -418,12 +392,10 @@ export const PIPELINE_DATA = {
418392 } ,
419393 ] ,
420394 modality : "cv" ,
421- color : "green" ,
422395 } ,
423396 "text-to-image" : {
424397 name : "Text-to-Image" ,
425398 modality : "cv" ,
426- color : "yellow" ,
427399 } ,
428400 "image-to-text" : {
429401 name : "Image-to-Text" ,
@@ -434,7 +406,6 @@ export const PIPELINE_DATA = {
434406 } ,
435407 ] ,
436408 modality : "cv" ,
437- color : "red" ,
438409 } ,
439410 "image-to-image" : {
440411 name : "Image-to-Image" ,
@@ -453,27 +424,22 @@ export const PIPELINE_DATA = {
453424 } ,
454425 ] ,
455426 modality : "cv" ,
456- color : "indigo" ,
457427 } ,
458428 "image-to-video" : {
459429 name : "Image-to-Video" ,
460430 modality : "cv" ,
461- color : "indigo" ,
462431 } ,
463432 "unconditional-image-generation" : {
464433 name : "Unconditional Image Generation" ,
465434 modality : "cv" ,
466- color : "green" ,
467435 } ,
468436 "video-classification" : {
469437 name : "Video Classification" ,
470438 modality : "cv" ,
471- color : "blue" ,
472439 } ,
473440 "reinforcement-learning" : {
474441 name : "Reinforcement Learning" ,
475442 modality : "rl" ,
476- color : "red" ,
477443 } ,
478444 robotics : {
479445 name : "Robotics" ,
@@ -488,7 +454,6 @@ export const PIPELINE_DATA = {
488454 name : "Task Planning" ,
489455 } ,
490456 ] ,
491- color : "blue" ,
492457 } ,
493458 "tabular-classification" : {
494459 name : "Tabular Classification" ,
@@ -503,7 +468,6 @@ export const PIPELINE_DATA = {
503468 name : "Tabular Multi Label Classification" ,
504469 } ,
505470 ] ,
506- color : "blue" ,
507471 } ,
508472 "tabular-regression" : {
509473 name : "Tabular Regression" ,
@@ -514,7 +478,6 @@ export const PIPELINE_DATA = {
514478 name : "Tabular Single Column Regression" ,
515479 } ,
516480 ] ,
517- color : "blue" ,
518481 } ,
519482 "tabular-to-text" : {
520483 name : "Tabular to Text" ,
@@ -525,13 +488,11 @@ export const PIPELINE_DATA = {
525488 name : "RDF to text" ,
526489 } ,
527490 ] ,
528- color : "blue" ,
529491 hideInModels : true ,
530492 } ,
531493 "table-to-text" : {
532494 name : "Table to Text" ,
533495 modality : "nlp" ,
534- color : "blue" ,
535496 hideInModels : true ,
536497 } ,
537498 "multiple-choice" : {
@@ -547,13 +508,11 @@ export const PIPELINE_DATA = {
547508 } ,
548509 ] ,
549510 modality : "nlp" ,
550- color : "blue" ,
551511 hideInModels : true ,
552512 } ,
553513 "text-ranking" : {
554514 name : "Text Ranking" ,
555515 modality : "nlp" ,
556- color : "red" ,
557516 } ,
558517 "text-retrieval" : {
559518 name : "Text Retrieval" ,
@@ -576,7 +535,6 @@ export const PIPELINE_DATA = {
576535 } ,
577536 ] ,
578537 modality : "nlp" ,
579- color : "indigo" ,
580538 hideInModels : true ,
581539 } ,
582540 "time-series-forecasting" : {
@@ -592,17 +550,14 @@ export const PIPELINE_DATA = {
592550 name : "Multivariate Time Series Forecasting" ,
593551 } ,
594552 ] ,
595- color : "blue" ,
596553 } ,
597554 "text-to-video" : {
598555 name : "Text-to-Video" ,
599556 modality : "cv" ,
600- color : "green" ,
601557 } ,
602558 "image-text-to-text" : {
603559 name : "Image-Text-to-Text" ,
604560 modality : "multimodal" ,
605- color : "red" ,
606561 hideInDatasets : true ,
607562 } ,
608563 "visual-question-answering" : {
@@ -614,7 +569,6 @@ export const PIPELINE_DATA = {
614569 } ,
615570 ] ,
616571 modality : "multimodal" ,
617- color : "red" ,
618572 } ,
619573 "document-question-answering" : {
620574 name : "Document Question Answering" ,
@@ -625,48 +579,39 @@ export const PIPELINE_DATA = {
625579 } ,
626580 ] ,
627581 modality : "multimodal" ,
628- color : "blue" ,
629582 hideInDatasets : true ,
630583 } ,
631584 "zero-shot-image-classification" : {
632585 name : "Zero-Shot Image Classification" ,
633586 modality : "cv" ,
634- color : "yellow" ,
635587 } ,
636588 "graph-ml" : {
637589 name : "Graph Machine Learning" ,
638590 modality : "other" ,
639- color : "green" ,
640591 } ,
641592 "mask-generation" : {
642593 name : "Mask Generation" ,
643594 modality : "cv" ,
644- color : "indigo" ,
645595 } ,
646596 "zero-shot-object-detection" : {
647597 name : "Zero-Shot Object Detection" ,
648598 modality : "cv" ,
649- color : "yellow" ,
650599 } ,
651600 "text-to-3d" : {
652601 name : "Text-to-3D" ,
653602 modality : "cv" ,
654- color : "yellow" ,
655603 } ,
656604 "image-to-3d" : {
657605 name : "Image-to-3D" ,
658606 modality : "cv" ,
659- color : "green" ,
660607 } ,
661608 "image-feature-extraction" : {
662609 name : "Image Feature Extraction" ,
663610 modality : "cv" ,
664- color : "indigo" ,
665611 } ,
666612 "video-text-to-text" : {
667613 name : "Video-Text-to-Text" ,
668614 modality : "multimodal" ,
669- color : "blue" ,
670615 hideInDatasets : false ,
671616 } ,
672617 "keypoint-detection" : {
@@ -678,25 +623,21 @@ export const PIPELINE_DATA = {
678623 } ,
679624 ] ,
680625 modality : "cv" ,
681- color : "red" ,
682626 hideInDatasets : true ,
683627 } ,
684628 "visual-document-retrieval" : {
685629 name : "Visual Document Retrieval" ,
686630 modality : "multimodal" ,
687- color : "yellow" ,
688631 hideInDatasets : true ,
689632 } ,
690633 "any-to-any" : {
691634 name : "Any-to-Any" ,
692635 modality : "multimodal" ,
693- color : "yellow" ,
694636 hideInDatasets : true ,
695637 } ,
696638 other : {
697639 name : "Other" ,
698640 modality : "other" ,
699- color : "blue" ,
700641 hideInModels : true ,
701642 hideInDatasets : true ,
702643 } ,
0 commit comments