shivam-1706 commited on
Commit
5376c03
·
verified ·
1 Parent(s): db58110

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -48,7 +48,7 @@ def load_sentiment_model():
48
 
49
  # Verify model has 5 classes
50
  if model.config.num_labels == 5:
51
- st.success("✅ 5-class DistilBERT model loaded successfully!")
52
  else:
53
  st.warning(f"⚠️ Model has {model.config.num_labels} classes, expected 5")
54
 
@@ -75,7 +75,7 @@ with st.spinner("Loading DistilBERT model..."):
75
  # Load LLM client automatically
76
  llm_client = load_llm_client()
77
  if llm_client:
78
- st.success("✅ GitHub Models API connected automatically!")
79
 
80
  def fix_problematic_words(text):
81
  """Quick fix for specific words causing classification issues"""
 
48
 
49
  # Verify model has 5 classes
50
  if model.config.num_labels == 5:
51
+ st.success("✅ 5-class sentiment model loaded successfully!")
52
  else:
53
  st.warning(f"⚠️ Model has {model.config.num_labels} classes, expected 5")
54
 
 
75
  # Load LLM client automatically
76
  llm_client = load_llm_client()
77
  if llm_client:
78
+ st.success("✅ GPT4.0 mini connected automatically!")
79
 
80
  def fix_problematic_words(text):
81
  """Quick fix for specific words causing classification issues"""