Hugging Face's logo

Spaces:
ashishkgpian
/
symptoms-to-disease
Sleeping

App Files Community
ashishkgpian commited on
Commit
a7ec3bd
·
verified ·
1 Parent(s): 05326b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  # Load the model
5
  classifier = pipeline(
6
  "text-classification",
7
- model="ashishkgpian/biolinkbert_base_icd9_classifier_ehr_symptoms_text_icd9_150_epochs"
8
  )
9
 
10
  def classify_symptoms(text):
 
4
  # Load the model
5
  classifier = pipeline(
6
  "text-classification",
7
+ model="ashishkgpian/biobert_icd9_classifier_ehr"
8
  )
9
 
10
  def classify_symptoms(text):