Ryan commited on
Commit
15b45ea
·
1 Parent(s): a1403d1
Files changed (1) hide show
  1. processors/roberta_analysis.py +10 -2
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
- score1 = sentiment_results[0]["sentiment_score"]
223
- score2 = sentiment_results[1]["sentiment_score"]
 
 
 
 
 
 
 
 
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)