Pfaffencode commited on
Commit
c7937d8
·
verified ·
1 Parent(s): d700342

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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="DistilBERT 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 temporary 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")
 
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")