Update src/components/model_nlp_intent.py

src/components/model_nlp_intent.py

@@ -4,21 +4,21 @@ from huggingface_hub import snapshot_download
 import joblib
 import os
 
-# Download the entire repository once at startup
-print("Downloading model from Hugging Face...")
+
+
 repo_path = snapshot_download(
     repo_id="samithcs/nlp_intent_model",
     repo_type="model"
 )
-print(f"Model downloaded to: {repo_path}")
 
-# Construct paths to subdirectories
+
+
 MODEL_PATH = os.path.join(repo_path, "nlp_intent", "intent_model")
 TOKENIZER_PATH = os.path.join(repo_path, "nlp_intent", "intent_tokenizer")
 LABEL_PATH = os.path.join(repo_path, "nlp_intent", "label_encoder.joblib")
 
-# Load model, tokenizer, and label encoder from local paths
-model = TFDistilBertForSequenceClassification.from_pretrained(MODEL_PATH, from_tf=True)
+
+model = TFDistilBertForSequenceClassification.from_pretrained(MODEL_PATH)
 tokenizer = DistilBertTokenizer.from_pretrained(TOKENIZER_PATH)
 label_encoder = joblib.load(LABEL_PATH)