edit

app.py

@@ -33,8 +33,15 @@ def siebert(data):
     label = result[0]['label']
     score = result[0]['score']
 
-    return label, score 
+    return label, score
 
+def finetuned(data):
+    specific_model = pipeline(model='dahongj/finetuned_toxictweets')
+    result = specific_model(data)
+    label = result[0]['label']
+    score = result[0]['score']
+
+    return label, score
 
 def getSent(data, model):
     if(model == 'Bertweet'):
@@ -52,6 +59,11 @@ def getSent(data, model):
         col1, col2 = st.columns(2)
         col1.metric("Feeling",label,None)
         col2.metric("Score",score,None)
+    elif(model == 'Finetuned'):
+        label, score = finetuned(data)
+        col1, col2 = st.columns(2)
+        col1.metric(label,label,None)
+        col2.metric("Score",score,None)
 
 def rendPage():
     st.title("Sentiment Analysis")
@@ -59,7 +71,7 @@ def rendPage():
     st.text("")
     type = st.selectbox(
         'Choose your model',
-        ('Bertweet','Roberta','Siebert'))
+        ('Bertweet','Roberta','Siebert','Finetuned'))
     st.text("")
 
     if st.button('Calculate'):