Update app.py

app.py

@@ -11,9 +11,9 @@ models = {
 
 def textblob_sentiment(text):
     polarity = TextBlob(text).sentiment.polarity
-    if polarity > 0:
+    if polarity > 0.2:
         return 'Positive'
-    elif polarity < 0:
+    elif polarity < -0.2:
         return 'Negative'
     else:
         return 'Neutral'
@@ -21,9 +21,9 @@ def textblob_sentiment(text):
 def vader_sentiment(text):
     analyzer = SentimentIntensityAnalyzer()
     scores = analyzer.polarity_scores(text)
-    if scores['compound'] > 0:
+    if scores['compound'] > 0.5:
         return 'Positive'
-    elif scores['compound'] < 0:
+    elif scores['compound'] < -0.5:
         return 'Negative'
     else:
         return 'Neutral'
@@ -42,7 +42,13 @@ if st.button('Predict'):
         st.write(f'Sentiment: {sentiment}')
     else:
         model = models[model_name]
-        result = model(text)[0]
-        label = result['label']
-        score = result['score']
-        st.write(f'Sentiment: {label} (Score: {score:.2f})')
+        results = model(text, return_all_scores=True)
+        st.write('Results:')
+        for result in results:
+            label = result['label']
+            score = result['score']
+            if score > 0.2:
+                st.write(f'{label} (Score: {score:.2f})')
+        mixed_score = 1 - sum([r['score'] for r in results])
+        if mixed_score > 0.2:
+            st.write(f'Mixed (Score: {mixed_score:.2f})')