kenlkehl commited on
Commit
784ed65
·
verified ·
1 Parent(s): f8673db

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -10,7 +10,7 @@
10
  # Set to None to skip auto-loading, or provide model path/HuggingFace ID
11
  MODEL_CONFIG = {
12
  # TinyBERT tagger for extracting relevant excerpts
13
- "tagger": "ksg-dfci/TinyOncoBertTagger-1225", # e.g., "prajjwal1/bert-tiny" or "./auto-tiny-bert-tagger"
14
 
15
  # Sentence transformer for embedding patient summaries and trials
16
  "embedder": "ksg-dfci/TrialSpace-1225", # e.g., "Qwen/Qwen3-Embedding-0.6B" or "./reranker_round2.model"
 
10
  # Set to None to skip auto-loading, or provide model path/HuggingFace ID
11
  MODEL_CONFIG = {
12
  # TinyBERT tagger for extracting relevant excerpts
13
+ "tagger": "ksg-dfci/TinyBertOncoTagger-1225", # e.g., "prajjwal1/bert-tiny" or "./auto-tiny-bert-tagger"
14
 
15
  # Sentence transformer for embedding patient summaries and trials
16
  "embedder": "ksg-dfci/TrialSpace-1225", # e.g., "Qwen/Qwen3-Embedding-0.6B" or "./reranker_round2.model"