dejanseo commited on
Commit
29fac90
·
verified ·
1 Parent(s): a043951

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +12 -1
src/streamlit_app.py CHANGED
@@ -18,6 +18,17 @@ model_name = 'dejanseo/sentiment'
18
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
19
  tokenizer = AutoTokenizer.from_pretrained(model_name)
20
 
 
 
 
 
 
 
 
 
 
 
 
21
  # Sentiment labels as textual descriptions
22
  sentiment_labels = {
23
  0: "very positive",
@@ -82,7 +93,7 @@ def classify_sentences(text):
82
  return sentence_scores
83
 
84
  # Streamlit UI
85
- st.title("Sentiment Classification Model by DEJAN")
86
 
87
  url = st.text_input("Enter URL:")
88
 
 
18
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
19
  tokenizer = AutoTokenizer.from_pretrained(model_name)
20
 
21
+ st.set_page_config(
22
+ page_title="Sentiment Analysis Tool by DEJAN AI",
23
+ page_icon="🔎",
24
+ layout="wide"
25
+ )
26
+
27
+ st.logo(
28
+ image="https://dejan.ai/wp-content/uploads/2024/02/dejan-300x103.png",
29
+ link="https://dejan.ai/",
30
+ )
31
+
32
  # Sentiment labels as textual descriptions
33
  sentiment_labels = {
34
  0: "very positive",
 
93
  return sentence_scores
94
 
95
  # Streamlit UI
96
+ st.title("Sentiment Classification Model")
97
 
98
  url = st.text_input("Enter URL:")
99