From 734e70bebe2256bc3173e7938d3e750f36001e89 Mon Sep 17 00:00:00 2001
From: Pavel Basov <pbasov@mirantis.com>
Date: Wed, 28 Feb 2024 19:59:25 +0100
Subject: [PATCH] Add HuggingFace predictor type support

---
 frontend/src/app/shared/utils.ts            | 3 +++
 frontend/src/app/types/kfserving/v1beta1.ts | 2 ++
 2 files changed, 5 insertions(+)

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;
 }