Update app.py
Browse files
app.py
CHANGED
@@ -76,7 +76,7 @@ class SpeechAnalyzer:
|
|
76 |
self.ner_model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
|
77 |
self.ner_pipeline = pipeline("ner", model=self.ner_model, tokenizer=self.ner_tokenizer)
|
78 |
|
79 |
-
def split_text(self, text, max_length=
|
80 |
"""Split long text into overlapping segments"""
|
81 |
words = text.split()
|
82 |
segments = []
|
@@ -128,7 +128,7 @@ class SpeechAnalyzer:
|
|
128 |
|
129 |
def analyze_emotional_trajectory(self, text, window_size=5):
|
130 |
"""Enhanced emotional trajectory analysis using sentence-level processing"""
|
131 |
-
segments = self.split_text(text, max_length=
|
132 |
sentiment_scores = []
|
133 |
|
134 |
for segment in segments:
|
@@ -315,7 +315,7 @@ def main():
|
|
315 |
col1, col2 = st.columns(2)
|
316 |
|
317 |
# Create consistent segments for both analyses
|
318 |
-
segments = analyzer.split_text(text, max_length=
|
319 |
num_segments = len(segments)
|
320 |
segment_labels = [f"{i+1}" for i in range(num_segments)]
|
321 |
|
|
|
76 |
self.ner_model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
|
77 |
self.ner_pipeline = pipeline("ner", model=self.ner_model, tokenizer=self.ner_tokenizer)
|
78 |
|
79 |
+
def split_text(self, text, max_length=512):
|
80 |
"""Split long text into overlapping segments"""
|
81 |
words = text.split()
|
82 |
segments = []
|
|
|
128 |
|
129 |
def analyze_emotional_trajectory(self, text, window_size=5):
|
130 |
"""Enhanced emotional trajectory analysis using sentence-level processing"""
|
131 |
+
segments = self.split_text(text, max_length=512)
|
132 |
sentiment_scores = []
|
133 |
|
134 |
for segment in segments:
|
|
|
315 |
col1, col2 = st.columns(2)
|
316 |
|
317 |
# Create consistent segments for both analyses
|
318 |
+
segments = analyzer.split_text(text, max_length=512)
|
319 |
num_segments = len(segments)
|
320 |
segment_labels = [f"{i+1}" for i in range(num_segments)]
|
321 |
|