Update README.md
Browse files
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
|