Commit
·
2f3c934
1
Parent(s):
78616fa
fixed indent issue
Browse files
app.py
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
# Load the BioClinicalBERT model for IBD classification
|
|
|
5 |
"text-classification",
|
6 |
model="MattStammers/BioClinicalBERT_IBD",
|
7 |
return_all_scores=True
|
@@ -51,5 +52,5 @@ demo = gr.Interface(
|
|
51 |
description="Classify free-text clinical documents as IBD vs non-IBD using BioClinicalBERT."
|
52 |
)
|
53 |
|
54 |
-
# Launch on Hugging Face Spaces
|
55 |
demo.launch(share=True)
|
|
|
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/BioClinicalBERT_IBD",
|
8 |
return_all_scores=True
|
|
|
52 |
description="Classify free-text clinical documents as IBD vs non-IBD using BioClinicalBERT."
|
53 |
)
|
54 |
|
55 |
+
# Launch publicly on Hugging Face Spaces
|
56 |
demo.launch(share=True)
|