admesh
/

agentic-intent-classifier

Text Classification

intent-classification

conversational-ai

calibrated-confidence

text-embeddings-inference

Model card Files Files and versions

manikumargouni commited on 13 days ago

Commit

3b3490c

·

verified ·

1 Parent(s): 43da358

Upload model_runtime.py with huggingface_hub

Files changed (1) hide show

model_runtime.py +6 -2

model_runtime.py CHANGED Viewed

@@ -12,8 +12,12 @@ os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-from config import HEAD_CONFIGS, HeadConfig, _looks_like_local_hf_model_dir
-from multitask_runtime import MultiTaskHeadProxy
 _TRAIN_SCRIPT_HINTS: dict[str, str] = {
     "intent_type": "python3 training/train.py",

 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+try:
+    from .config import HEAD_CONFIGS, HeadConfig, _looks_like_local_hf_model_dir  # type: ignore
+    from .multitask_runtime import MultiTaskHeadProxy  # type: ignore
+except ImportError:
+    from config import HEAD_CONFIGS, HeadConfig, _looks_like_local_hf_model_dir
+    from multitask_runtime import MultiTaskHeadProxy
 _TRAIN_SCRIPT_HINTS: dict[str, str] = {
     "intent_type": "python3 training/train.py",