Update src/streamlit_app.py
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
|
@@ -5,9 +5,9 @@ import streamlit as st
|
|
| 5 |
import torch
|
| 6 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 7 |
|
| 8 |
-
st.set_page_config(page_title="
|
| 9 |
|
| 10 |
-
st.warning("⚙️ **Model Update in Progress** – The classification model is currently being updated. You may experience
|
| 11 |
|
| 12 |
# Resolve default model id from env or use your public repo id
|
| 13 |
DEFAULT_MODEL_ID = os.environ.get("HF_MODEL_ID", "Pfaffencode/Clinical-ModernBERT-Fine-Tuned-MH")
|
|
|
|
| 5 |
import torch
|
| 6 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 7 |
|
| 8 |
+
st.set_page_config(page_title="ModernBERT Mental Health Classifier", page_icon="🧠", layout="centered")
|
| 9 |
|
| 10 |
+
st.warning("⚙️ **Model Update in Progress** – The classification model is currently being updated. You may experience inconsistencies in predictions.")
|
| 11 |
|
| 12 |
# Resolve default model id from env or use your public repo id
|
| 13 |
DEFAULT_MODEL_ID = os.environ.get("HF_MODEL_ID", "Pfaffencode/Clinical-ModernBERT-Fine-Tuned-MH")
|