Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ def highlight(sentence):
|
|
48 |
st.markdown(highlighted_text, unsafe_allow_html=True)
|
49 |
|
50 |
#Load the trained model
|
51 |
-
with open(
|
52 |
rf = pickle.load(f)
|
53 |
# Load the BioBERT model and tokenizer
|
54 |
model_name = "dmis-lab/biobert-base-cased-v1.1"
|
|
|
48 |
st.markdown(highlighted_text, unsafe_allow_html=True)
|
49 |
|
50 |
#Load the trained model
|
51 |
+
with open("biobert_rf.pkl", 'rb') as f:
|
52 |
rf = pickle.load(f)
|
53 |
# Load the BioBERT model and tokenizer
|
54 |
model_name = "dmis-lab/biobert-base-cased-v1.1"
|