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

env hf token

Browse Source
This commit is contained in:
Tryndaron
2024-04-03 16:10:15 +02:00
parent 2b4819c68f
commit 018a2bd1cf
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_autotranscript.py
+5 -1
View File
@@ -2,7 +2,7 @@ import pytest
import torch import torch
from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
import os
""" """
@pytest.fixture @pytest.fixture
@@ -13,10 +13,14 @@ def example_audio_file(audio_test_2.mp4):
@pytest.fixture @pytest.fixture
def create_scraibe_instance(): def create_scraibe_instance():
if "HF_TOKEN" in os.environ:
return Scraibe(use_auth_token=os.environ["HF_TOKEN"] )
else:
return Scraibe() return Scraibe()
def test_scraibe_init(create_scraibe_instance): def test_scraibe_init(create_scraibe_instance):
model = create_scraibe_instance model = create_scraibe_instance
assert isinstance(model.transcriber, Transcriber) assert isinstance(model.transcriber, Transcriber)