hylee719
/

transcript-analysis-testing

Model card Files Files and versions Community

hylee commited on Nov 5, 2023

Commit

f9ef8dc

·

1 Parent(s): 337b02e

small fix

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -351,7 +351,7 @@ class EndpointHandler():
         question_model = QuestionModel(
             self.device, self.tokenizer, self.input_builder)
         question_model.run_inference(transcript)
-        transcript.update_utterance_roles()
         transcript.calculate_aggregate_word_count()
         talk_dist, talk_len = transcript.get_talk_distribution_and_length(uptake_speaker)
         talk_timeline = transcript.get_talk_timeline()

         question_model = QuestionModel(
             self.device, self.tokenizer, self.input_builder)
         question_model.run_inference(transcript)
+        transcript.update_utterance_roles(uptake_speaker)
         transcript.calculate_aggregate_word_count()
         talk_dist, talk_len = transcript.get_talk_distribution_and_length(uptake_speaker)
         talk_timeline = transcript.get_talk_timeline()