edit

app.py

@@ -27,6 +27,21 @@ def roberta(data):
 
     return label, score 
 
+def siebert(data):
+    specific_model = pipeline(model='siebert/sentiment-roberta-large-english')
+    result = specific_model(data)
+    label = result[0]['label']
+    score = result[0]['score']
+
+    if(label == 'LABEL_0'):
+        label = 'Negative'
+    elif(label == 'LABEL_1'):
+        label = 'Neutral'
+    else:
+        label = 'Positive'
+
+    return label, score 
+
 def getSent(data, model):
     if(model == 'Bertweet'):
         label,score = bertweet(data)
@@ -38,6 +53,11 @@ def getSent(data, model):
         col1, col2 = st.columns(2)
         col1.metric("Feeling",label,None)
         col2.metric("Score",score,None)
+    elif(model == 'Siebert'):
+        label,score = siebert(data)
+        col1, col2 = st.columns(2)
+        col1.metric("Feeling",label,None)
+        col2.metric("Score",score,None)
 
 def rendPage():
     st.title("Sentiment Analysis")
@@ -45,7 +65,7 @@ def rendPage():
     st.text("")
     type = st.selectbox(
         'Choose your model',
-        ('Bertweet','Roberta',))
+        ('Bertweet','Roberta','Siebert'))
     st.text("")
 
     if st.button('Calculate'):