dhruv2842 commited on
Commit
9c80776
·
verified ·
1 Parent(s): 739a4b7

Update utils/specialist_predictor.py

Browse files
Files changed (1) hide show
  1. utils/specialist_predictor.py +2 -1
utils/specialist_predictor.py CHANGED
@@ -6,7 +6,8 @@ import os
6
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
7
  # Load model components once
8
  bundle = joblib.load("semantic_specialist_model.pkl")
9
- model = SentenceTransformer(bundle["model_name"])
 
10
  known_embeddings = bundle["known_embeddings"]
11
  symptom_specialist_pairs = bundle["symptom_specialist_pairs"]
12
 
 
6
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
7
  # Load model components once
8
  bundle = joblib.load("semantic_specialist_model.pkl")
9
+ local_model_path = "models/all-MiniLM-L6-v2"
10
+ model = SentenceTransformer(local_model_path)
11
  known_embeddings = bundle["known_embeddings"]
12
  symptom_specialist_pairs = bundle["symptom_specialist_pairs"]
13