Spaces:
Sleeping
Sleeping
Ryan
commited on
Commit
·
15b45ea
1
Parent(s):
a1403d1
update
Browse files
processors/roberta_analysis.py
CHANGED
@@ -219,8 +219,16 @@ def compare_sentiment_roberta(texts, model_names=None):
|
|
219 |
# Compare sentiment scores
|
220 |
if len(sentiment_results) >= 2:
|
221 |
model1_name, model2_name = model_names[0], model_names[1]
|
222 |
-
|
223 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
224 |
|
225 |
# Calculate difference and determine which is more positive/negative
|
226 |
difference = abs(score1 - score2)
|
|
|
219 |
# Compare sentiment scores
|
220 |
if len(sentiment_results) >= 2:
|
221 |
model1_name, model2_name = model_names[0], model_names[1]
|
222 |
+
|
223 |
+
# Add null checks for the sentiment results
|
224 |
+
score1 = 0
|
225 |
+
score2 = 0
|
226 |
+
|
227 |
+
if sentiment_results[0] and "sentiment_score" in sentiment_results[0]:
|
228 |
+
score1 = sentiment_results[0]["sentiment_score"]
|
229 |
+
|
230 |
+
if sentiment_results[1] and "sentiment_score" in sentiment_results[1]:
|
231 |
+
score2 = sentiment_results[1]["sentiment_score"]
|
232 |
|
233 |
# Calculate difference and determine which is more positive/negative
|
234 |
difference = abs(score1 - score2)
|