rocky250 commited on
Commit
58b1bc8
·
verified ·
1 Parent(s): 9a97175

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -2
src/streamlit_app.py CHANGED
@@ -228,7 +228,8 @@ def load_models():
228
  "BanglaBERT": "rocky250/Sentiment-banglabert",
229
  "mBERT": "rocky250/Sentiment-mbert",
230
  "B-Base": "rocky250/Sentiment-bbase",
231
- "XLM-R": "rocky250/Sentiment-xlmr"
 
232
  }
233
 
234
  for name, repo in standard_models.items():
@@ -314,7 +315,7 @@ with col1:
314
  with col2:
315
  st.markdown("<div style='height: 20px'></div>", unsafe_allow_html=True)
316
  mode = st.radio("Analysis Mode:",
317
- ["Single Model", "Ensemble (Recommended)"],
318
  horizontal=True)
319
 
320
  model_options = {f"({i+1}) {name}": name for i, name in enumerate(models_dict.keys())}
 
228
  "BanglaBERT": "rocky250/Sentiment-banglabert",
229
  "mBERT": "rocky250/Sentiment-mbert",
230
  "B-Base": "rocky250/Sentiment-bbase",
231
+ "XLM-R": "rocky250/Sentiment-xlmr",
232
+ "Bangla-P": "rocky250/bangla-political"
233
  }
234
 
235
  for name, repo in standard_models.items():
 
315
  with col2:
316
  st.markdown("<div style='height: 20px'></div>", unsafe_allow_html=True)
317
  mode = st.radio("Analysis Mode:",
318
+ ["Single Model", "Ensemble"],
319
  horizontal=True)
320
 
321
  model_options = {f"({i+1}) {name}": name for i, name in enumerate(models_dict.keys())}