Spaces:
Build error
Build error
Update src/streamlit_app.py
Browse files- 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
|
| 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 |
|