narainkumbari commited on
Commit
8bb7e48
·
1 Parent(s): 17cfcdd

Fix: Load CPU-safe model for HF Space

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import io
9
 
10
  # Load model and tokenizer from local fine-tuned directory
11
  # Define base and adapter model paths
12
- BASE_MODEL = "stanfordai/BioMedLM" # or the path you used originally
13
  ADAPTER_PATH = "Tufan1/BioMedLM-Cardio-Fold4-CPU"
14
  tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
15
 
 
9
 
10
  # Load model and tokenizer from local fine-tuned directory
11
  # Define base and adapter model paths
12
+ BASE_MODEL = "stanford-crfm/BioMedLM" # or the path you used originally
13
  ADAPTER_PATH = "Tufan1/BioMedLM-Cardio-Fold4-CPU"
14
  tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
15