Hasitha16 commited on
Commit
0f55ff6
·
verified ·
1 Parent(s): 05dad7c

Update frontend.py

Browse files
Files changed (1) hide show
  1. frontend.py +2 -2
frontend.py CHANGED
@@ -96,7 +96,7 @@ with tab1:
96
  try:
97
  payload = {
98
  "text": st.session_state.review,
99
- "model": sentiment_model if sentiment_model != "Auto-detect" else None,
100
  "industry": industry,
101
  "product_category": product_category,
102
  "verbosity": verbosity,
@@ -183,7 +183,7 @@ with tab2:
183
  lines = [l.strip() for l in bulk_input.strip().splitlines() if l.strip()]
184
  payload = {
185
  "reviews": lines,
186
- "model": sentiment_model if sentiment_model != "Auto-detect" else None,
187
  "industry": None,
188
  "product_category": None,
189
  "device": None,
 
96
  try:
97
  payload = {
98
  "text": st.session_state.review,
99
+ "model": "distilbert-base-uncased-finetuned-sst-2-english" if sentiment_model == "Auto-detect" else sentiment_model,
100
  "industry": industry,
101
  "product_category": product_category,
102
  "verbosity": verbosity,
 
183
  lines = [l.strip() for l in bulk_input.strip().splitlines() if l.strip()]
184
  payload = {
185
  "reviews": lines,
186
+ "model": "distilbert-base-uncased-finetuned-sst-2-english" if sentiment_model == "Auto-detect" else sentiment_model,
187
  "industry": None,
188
  "product_category": None,
189
  "device": None,