First version of space

app.py

@@ -0,0 +1,51 @@
+import streamlit as st
+from transformers import pipeline
+
+
+label2stars = {
+    1: "⭐️",
+    2: "⭐️⭐️",
+    3: "⭐️⭐️⭐️",
+    4: "⭐️⭐️⭐️⭐️",
+    5: "⭐️⭐️⭐️⭐️⭐️",
+    "1 star": "⭐️",
+    "2 stars": "⭐️⭐️",
+    "3 stars": "⭐️⭐️⭐️",
+    "4 stars": "⭐️⭐️⭐️⭐️",
+    "5 stars": "⭐️⭐️⭐️⭐️⭐️",
+}
+
+@st.cache_resource
+def load_model(model_id):
+    return pipeline("sentiment-analysis", model=model_id)
+
+
+bert_model = load_model("nlptown/bert-base-multilingual-uncased-sentiment")
+modernbert_model = load_model("nlptown/ModernBERT-base-sentiment-20241228")
+
+st.title("NLP Town Sentiment Analysis")
+
+example1 = "I have sensitive eyes and this is non-irritating. My eyes water a lot and that often leads to product getting into my eyes and causing stinging and or burning in the eye area. This cream has not bothered me at all."
+example2 = "Very natural, light weight and good quality. I use on brides who are not used to lashes but still want a bit of length and definition."
+example3 = "I have to admit that I'm very new to meditation and guided imagery, but I found the suggested imagery of a \"guardian\" closing the drapes in my bedroom and then sitting next to my bed while I was sleeping to be unnerving. I was wide awake at the end of the CD and was completely disappointed with this purchase. I'm glad that I am able to return it."
+example4 = "Catchy promises, little delivery."
+
+option = st.selectbox("Examples", [example1, example2, example3, example4], index=None, placeholder='Select an example or enter your text below')
+
+query = st.text_area("Enter your text here", value=option)
+click = st.button("Analyze", type="primary")
+
+if query or click:
+
+    bert_result = bert_model(query)[0]
+    modernbert_result = modernbert_model(query)[0]
+
+    col1, col2 = st.columns([3, 1])
+
+    with col1:
+        st.write("[nlptown/bert-base-multilingual-uncased-sentiment](https://huggingface.co/nlptown/bert-base-multilingual-uncased-sentiment):")
+        st.write("[nlptown/ModernBERT-base-sentiment](https://www.nlp.town):")
+
+    with col2:
+        st.write(label2stars[bert_result['label']])
+        st.write(label2stars[modernbert_result['label']])