diff --git a/python/paddle_serving_app/models/model_list.py b/python/paddle_serving_app/models/model_list.py index a2019997968ce21a30669b2acd1421355b1e0fdd..ca33933b7797dc0bd1cb5881e73ba3fc9d82f3c1 100644 --- a/python/paddle_serving_app/models/model_list.py +++ b/python/paddle_serving_app/models/model_list.py @@ -25,7 +25,9 @@ class ServingModels(object): self.model_dict["SemanticRepresentation"] = ["ernie_base"] self.model_dict["ChineseWordSegmentation"] = ["lac"] self.model_dict["ObjectDetection"] = ["faster_rcnn", "yolov3"] - self.model_dict["ImageSegmentation"] = ["unet", "deeplabv3"] + self.model_dict["ImageSegmentation"] = [ + "unet", "deeplabv3", "mobilenet_cityspaces" + ] self.model_dict["ImageClassification"] = [ "resnet_v2_50_imagenet", "mobilenet_v2_imagenet" ]