diff --git a/frontend/src/app/shared/utils.ts b/frontend/src/app/shared/utils.ts index 24eae8c..cb0b819 100644 --- a/frontend/src/app/shared/utils.ts +++ b/frontend/src/app/shared/utils.ts @@ -234,6 +234,9 @@ export function getPredictorRuntime(predictor: PredictorSpec): string { if (predictorType === PredictorType.Lightgbm) { return 'LightGBM ModelServer'; } + if (predictorType === PredictorType.Huggingface) { + return 'HuggingFace ModelServer'; + } if (predictorType === PredictorType.Custom) { return 'Custom ModelServer'; } diff --git a/frontend/src/app/types/kfserving/v1beta1.ts b/frontend/src/app/types/kfserving/v1beta1.ts index 5e175fd..dee3148 100644 --- a/frontend/src/app/types/kfserving/v1beta1.ts +++ b/frontend/src/app/types/kfserving/v1beta1.ts @@ -47,6 +47,7 @@ export enum PredictorType { Xgboost = 'xgboost', Pmml = 'pmml', Lightgbm = 'lightgbm', + Huggingface = 'huggingface', Custom = 'custom', } @@ -59,6 +60,7 @@ export interface PredictorSpec extends V1PodSpec, ComponentExtensionSpec { onnx?: PredictorExtensionSpec; pmml?: PredictorExtensionSpec; lightgbm?: PredictorExtensionSpec; + huggingface?: PredictorExtensionSpec; model?: ModelSpec; }