admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

added relative Path

Browse Source
This commit is contained in:
Schmieder, Jacob
2024-09-30 12:27:41 +00:00
parent df9c5109f3
commit 2adbfaef51
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_autotranscript.py
+4 -4
View File
@@ -6,7 +6,7 @@ import os
@pytest.fixture
def create_scraibe_instance():
if "HF_TOKEN" in os.environ:
return Scraibe(use_auth_token=os.environ["HF_TOKEN"])
return Scraibe(use_auth_token=os.environ["HF_TOKEN"], whisper_model= "tiny")
else:
return Scraibe()
@@ -19,19 +19,19 @@ def test_scraibe_init(create_scraibe_instance):
def test_scraibe_autotranscribe(create_scraibe_instance):
model = create_scraibe_instance
transcript = model.autotranscribe('test/audio_test_2.mp4')
transcript = model.autotranscribe('./test/audio_test_2.mp4')
assert isinstance(transcript, Transcript)
def test_scraibe_diarization(create_scraibe_instance):
model = create_scraibe_instance
diarisation_result = model.diarization('test/audio_test_2.mp4')
diarisation_result = model.diarization('./test/audio_test_2.mp4')
assert isinstance(diarisation_result, dict)
def test_scraibe_transcribe(create_scraibe_instance):
model = create_scraibe_instance
transcription_result = model.transcribe('test/audio_test_2.mp4')
transcription_result = model.transcribe('./test/audio_test_2.mp4')
assert isinstance(transcription_result, str)
test/test_transcriber.py
+2 -2
View File
@@ -69,12 +69,12 @@ def test_get_whisper_kwargs():
def test_whisper_transcribe(whisper_instance):
model = whisper_instance
# mocker.patch.object(transcriber_instance.model, 'transcribe', return_value={'Hello, World !'} )
transcript = model.transcribe('test/audio_test_2.mp4')
transcript = model.transcribe('./test/audio_test_2.mp4')
assert isinstance(transcript, str)
def test_faster_whisper_transcribe(faster_whisper_instance):
model = faster_whisper_instance
# mocker.patch.object(transcriber_instance.model, 'transcribe', return_value={'Hello, World !'} )
transcript = model.transcribe('test/audio_test_2.mp4')
transcript = model.transcribe('./test/audio_test_2.mp4')
assert isinstance(transcript, str)