Update README.md

README.md

@@ -74,7 +74,7 @@ config = AutoConfig.from_pretrained(repo_id, subfolder=subfolder)
 tokenizer = AutoTokenizer.from_pretrained(repo_id, subfolder=subfolder)
 
 # 2. Initialize Model Architecture
-model = BERTDiseaseClassifier(model_type=config._name_or_path, num_symps=len(config.id2label))
+model = BERTDiseaseClassifier(model_type="bert-large-uncased", num_symps=len(config.id2label))
 
 # 3. Load Weights
 weights_path = hf_hub_download(repo_id=repo_id, subfolder=subfolder, filename="pytorch_model.bin")