Hasitha16 commited on
Commit
7b53476
Β·
verified Β·
1 Parent(s): 79bc441

Update frontend.py

Browse files
Files changed (1) hide show
  1. frontend.py +3 -2
frontend.py CHANGED
@@ -57,7 +57,8 @@ with st.sidebar:
57
  backend_url = st.text_input("🌐 Backend URL", value="http://localhost:8000")
58
 
59
  sentiment_model = st.selectbox("πŸ“Š Sentiment Model", [
60
- "Auto-detect", "distilbert-base-uncased-finetuned-sst-2-english",
 
61
  "nlptown/bert-base-multilingual-uncased-sentiment"
62
  ])
63
  industry = st.selectbox("🏭 Industry", ["Auto-detect", "Generic", "E-commerce", "Healthcare", "Education"])
@@ -132,7 +133,7 @@ with tab1:
132
  if data:
133
  st.subheader("πŸ“Œ Summary")
134
  st.info(data["summary"])
135
- st.caption(f"🧠 Summary | {verbosity} response")
136
  st.markdown(f"**Context:** `{data['industry']}` | `{data['product_category']}` | `Web`")
137
 
138
  st.metric("πŸ“Š Sentiment", data["sentiment"]["label"], delta=f"{data['sentiment']['score']:.2%}")
 
57
  backend_url = st.text_input("🌐 Backend URL", value="http://localhost:8000")
58
 
59
  sentiment_model = st.selectbox("πŸ“Š Sentiment Model", [
60
+ "Auto-detect",
61
+ "distilbert-base-uncased-finetuned-sst-2-english",
62
  "nlptown/bert-base-multilingual-uncased-sentiment"
63
  ])
64
  industry = st.selectbox("🏭 Industry", ["Auto-detect", "Generic", "E-commerce", "Healthcare", "Education"])
 
133
  if data:
134
  st.subheader("πŸ“Œ Summary")
135
  st.info(data["summary"])
136
+ st.caption("🧠 Summary Model: facebook/bart-large-cnn | " + verbosity + " response")
137
  st.markdown(f"**Context:** `{data['industry']}` | `{data['product_category']}` | `Web`")
138
 
139
  st.metric("πŸ“Š Sentiment", data["sentiment"]["label"], delta=f"{data['sentiment']['score']:.2%}")