Upload pipeline.py with huggingface_hub

pipeline.py

@@ -42,6 +42,7 @@ Supported HF deployment surfaces
 
 from __future__ import annotations
 
+import importlib
 import sys
 from pathlib import Path
 from typing import Union
@@ -230,8 +231,14 @@ class AdmeshIntentPipeline(_HFPipeline):
                     stacklevel=2,
                 )
             else:
-                from multitask_runtime import get_multitask_runtime  # noqa: PLC0415
-                from model_runtime import get_head  # noqa: PLC0415
+                if __package__:
+                    get_multitask_runtime = importlib.import_module(
+                        f"{__package__}.multitask_runtime"
+                    ).get_multitask_runtime
+                    get_head = importlib.import_module(f"{__package__}.model_runtime").get_head
+                else:
+                    get_multitask_runtime = importlib.import_module("multitask_runtime").get_multitask_runtime
+                    get_head = importlib.import_module("model_runtime").get_head
 
                 rt = get_multitask_runtime()
                 if rt._model is not None:
@@ -296,8 +303,12 @@ class AdmeshIntentPipeline(_HFPipeline):
 
     def _ensure_loaded(self) -> None:
         if self._classify_fn is None:
-            from combined_inference import classify_query  # noqa: PLC0415
-            self._classify_fn = classify_query
+            if __package__:
+                self._classify_fn = importlib.import_module(
+                    f"{__package__}.combined_inference"
+                ).classify_query
+            else:
+                self._classify_fn = importlib.import_module("combined_inference").classify_query
 
     def __repr__(self) -> str:
         state = "loaded" if self._classify_fn is not None else "not yet loaded"