First draft

Author: NielsRogge

packages/tasks/src/tasks/image-segmentation/about.md

@@ -48,7 +48,7 @@ import { HfInference } from "@huggingface/inference";
 const inference = new HfInference(HF_TOKEN);
 await inference.imageSegmentation({
 	data: await (await fetch("https://picsum.photos/300/300")).blob(),
-	model: "facebook/detr-resnet-50-panoptic",
+	model: "facebook/mask2former-swin-base-coco-panoptic",
 });
 ```
 

packages/tasks/src/tasks/image-segmentation/data.ts

@@ -44,21 +44,21 @@ const taskData: TaskDataCustom = {
 	models: [
 		{
 			// TO DO: write description
-			description: "Solid panoptic segmentation model trained on the COCO 2017 benchmark dataset.",
-			id: "facebook/detr-resnet-50-panoptic",
+			description: "Solid semantic segmentation model fine-tuned on ADE20k, an important benchmark comprising 150 classes.",
+			id: "openmmlab/upernet-convnext-small",
 		},
 		{
 			description: "Background removal model.",
 			id: "briaai/RMBG-1.4",
 		},
-		{
-			description: "Semantic segmentation model trained on ADE20k benchmark dataset with 512x512 resolution.",
-			id: "nvidia/segformer-b0-finetuned-ade-512-512",
-		},
 		{
 			description: "A multipurpose image segmentation model for high resolution images.",
 			id: "ZhengPeng7/BiRefNet",
 		},
+		{
+			description: "Semantic segmentation model fine-tuned on ADE20k at 512x512 resolution.",
+			id: "nvidia/segformer-b0-finetuned-ade-512-512",
+		},
 		{
 			description: "Panoptic segmentation model trained COCO (common objects) dataset.",
 			id: "facebook/mask2former-swin-large-coco-panoptic",

packages/tasks/src/tasks/object-detection/data.ts

@@ -43,15 +43,15 @@ const taskData: TaskDataCustom = {
 	],
 	models: [
 		{
-			description: "Solid object detection model trained on the benchmark dataset COCO 2017.",
+			description: "Solid object detection model pre-trained on the COCO 2017 dataset.",
 			id: "facebook/detr-resnet-50",
 		},
 		{
-			description: "Strong object detection model trained on ImageNet-21k dataset.",
-			id: "microsoft/beit-base-patch16-224-pt22k-ft22k",
+			description: "YOLOv10 is great model for real-time and accurate object detection.",
+			id: "jameslahm/yolov10x",
 		},
 		{
-			description: "Fast and accurate object detection model trained on COCO dataset.",
+			description: "RT-DETR is a fast and accurate object detection model pre-trained on the Object365 dataset.",
 			id: "PekingU/rtdetr_r18vd_coco_o365",
 		},
 	],