shallowblueQAQ commited on
Commit
aaec136
·
verified ·
1 Parent(s): a54a086

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -92,6 +92,7 @@ config = AutoConfig.from_pretrained(repo_id, subfolder=subfolder)
92
  tokenizer = AutoTokenizer.from_pretrained(repo_id, subfolder=subfolder)
93
 
94
  # 2. Initialize Model Architecture
 
95
  model = BERTDiseaseClassifier(model_type="bert-large-uncased", num_symps=len(config.id2label))
96
 
97
  # 3. Load Weights
 
92
  tokenizer = AutoTokenizer.from_pretrained(repo_id, subfolder=subfolder)
93
 
94
  # 2. Initialize Model Architecture
95
+ # NOTE: If you are running offline, you can replace "bert-large-uncased" with your local path (e.g., "/path/to/bert-large-uncased").
96
  model = BERTDiseaseClassifier(model_type="bert-large-uncased", num_symps=len(config.id2label))
97
 
98
  # 3. Load Weights