kambris commited on
Commit
f3a40fd
·
verified ·
1 Parent(s): 22f4a12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=256, overlap=50):
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=256)
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=256)
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