kambris commited on
Commit
e601c14
·
verified ·
1 Parent(s): 3f1e4b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -298,18 +298,17 @@ def main():
298
 
299
  for segment in segments:
300
  result = analyzer.sentiment_pipeline(segment[:512])[0]
301
- sentiment_results.append(result)
302
-
303
- # Convert sentiment labels to numerical values
304
- sentiment_values = [1 if result['label'] == 'POSITIVE' else -1 for result in sentiment_results]
305
 
306
  # Create segment labels
307
- num_segments = len(sentiment_values)
308
  segment_labels = [f"{i+1}" for i in range(num_segments)]
309
 
310
  trajectory_fig = go.Figure(data=go.Scatter(
311
  x=segment_labels,
312
- y=sentiment_values,
313
  mode='lines+markers',
314
  line=dict(
315
  color='#1f77b4',
 
298
 
299
  for segment in segments:
300
  result = analyzer.sentiment_pipeline(segment[:512])[0]
301
+ # Convert score to a more granular value instead of binary
302
+ sentiment_value = result['score'] if result['label'] == 'POSITIVE' else -result['score']
303
+ sentiment_results.append(sentiment_value)
 
304
 
305
  # Create segment labels
306
+ num_segments = len(sentiment_results)
307
  segment_labels = [f"{i+1}" for i in range(num_segments)]
308
 
309
  trajectory_fig = go.Figure(data=go.Scatter(
310
  x=segment_labels,
311
+ y=sentiment_results,
312
  mode='lines+markers',
313
  line=dict(
314
  color='#1f77b4',