ashishkgpian commited on
Commit
8cbff0f
·
verified ·
1 Parent(s): f07a774

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  # Load the model
7
  classifier = pipeline(
8
  "text-classification",
9
- model="ashishkgpian/biolinkbert_base_icd9_classifier_ehr_symptoms_text_icd9_150_epochs_v2"
10
  )
11
 
12
  # Load ICD9 codes data
 
6
  # Load the model
7
  classifier = pipeline(
8
  "text-classification",
9
+ model="ashishkgpian/biobert_icd9_classifier_ehr_symptoms_text_icd9_150_epochs"
10
  )
11
 
12
  # Load ICD9 codes data