MattStammers commited on
Commit
a5d3d22
·
1 Parent(s): ffdbac5

updated text

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Load the BioClinicalBERT model for IBD classification
5
  classifier = pipeline(
6
  "text-classification",
7
  model="MattStammers/Distil_IBD_BERT",
@@ -58,7 +58,7 @@ demo = gr.Interface(
58
  [negative_example]
59
  ],
60
  title="IBD Cohort Identifier",
61
- description="Classify free-text clinical documents as IBD vs non-IBD using BioClinicalBERT."
62
  )
63
 
64
  # Launch publicly on Hugging Face Spaces
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Load the Distil_IBD_BERT model for IBD classification
5
  classifier = pipeline(
6
  "text-classification",
7
  model="MattStammers/Distil_IBD_BERT",
 
58
  [negative_example]
59
  ],
60
  title="IBD Cohort Identifier",
61
+ description="Classify free-text clinical documents as IBD vs non-IBD using Distil_IBD_BERT."
62
  )
63
 
64
  # Launch publicly on Hugging Face Spaces