Mirrors/Voice-Cloning-App
1
0
Fork 0
mirror of https://github.com/voice-cloning-app/Voice-Cloning-App.git synced 2026-08-29 10:09:19 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix create dataset CLI

Browse Source
This commit is contained in:
BenAAndrew
2021-07-31 14:13:11 +01:00
parent 1bbcd08fc8
commit 3ee50d4304
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dataset/create_dataset.py
View File

@@ -8,6 +8,7 @@ sys.path.append(dirname(dirname(abspath(__file__))))
from dataset.audio_processing import convert_audio
from dataset.clip_generator import clip_generator
from dataset.analysis import save_dataset_info
from dataset.transcribe import create_transcription_model
def create_dataset(
@@ -82,4 +83,4 @@ if __name__ == "__main__":
parser.add_argument("-i", "--info_path", help="Path to save info file", type=str, default="info.json")
args = parser.parse_args()
create_dataset(**vars(args))
create_dataset(**vars(args), transcription_model=create_transcription_model())

3
dataset/transcribe.py
View File

@@ -118,5 +118,6 @@ if __name__ == "__main__":
parser.add_argument("-i", "--input_path", help="Path to audio file", type=str, required=True)
args = parser.parse_args()
text = transcribe(args.input_path)
model = create_transcription_model()
text = model.transcribe(args.input_path)
print("Text: ", text)