Update app.py
Browse files
app.py
CHANGED
@@ -298,18 +298,17 @@ def main():
|
|
298 |
|
299 |
for segment in segments:
|
300 |
result = analyzer.sentiment_pipeline(segment[:512])[0]
|
301 |
-
|
302 |
-
|
303 |
-
|
304 |
-
sentiment_values = [1 if result['label'] == 'POSITIVE' else -1 for result in sentiment_results]
|
305 |
|
306 |
# Create segment labels
|
307 |
-
num_segments = len(
|
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=
|
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',
|