Merge pull request #19 from kadirnar/fix-type

Update requirements.txt and TypeError error has been fixed.
kadirnar · Nov 23, 2023 · 9ceb18b · 9ceb18b
2 parents bf7e10c + dc14c2c
commit 9ceb18b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,9 @@
-gradio==4.5.0
+gradio==4.7.1
+moviepy==1.0.3
+numpy==1.26.2
+pyannote.audio==3.1.0
 pytube==15.0.0
+Requests==2.31.0
 torch==2.1.0
+torchaudio==2.1.0
 transformers==4.35.2
-accelerate
-moviepy
-pyannote
diff --git a/whisperplus/__init__.py b/whisperplus/__init__.py
@@ -1,5 +1,6 @@
 from whisperplus.pipelines.summarization import TextSummarizationPipeline
 from whisperplus.pipelines.whisper import SpeechToTextPipeline
+from whisperplus.pipelines.whisper_diarize import ASRDiarizationPipeline
 from whisperplus.utils.download_utils import download_and_convert_to_mp3
 from whisperplus.utils.text_utils import format_speech_to_dialogue
 

diff --git a/whisperplus/utils/text_utils.py b/whisperplus/utils/text_utils.py
@@ -9,7 +9,7 @@ def format_speech_to_dialogue(speech_text):
         str: Formatted text in dialogue format.
     """
     # Parse the given text appropriately
-    dialog_list = eval(speech_text)
+    dialog_list = eval(str(speech_text))
     dialog_text = ""
 
     for i, turn in enumerate(dialog_list):