env hf token

This commit is contained in:
Tryndaron
2024-04-03 16:10:15 +02:00
parent 2b4819c68f
commit 018a2bd1cf
+6 -2
View File
@@ -2,7 +2,7 @@ import pytest
import torch
from scraibe import Scraibe, Diariser, Transcriber, Transcript, AudioProcessor
from unittest.mock import MagicMock, patch
import os
"""
@pytest.fixture
@@ -13,7 +13,11 @@ def example_audio_file(audio_test_2.mp4):
@pytest.fixture
def create_scraibe_instance():
return Scraibe()
if "HF_TOKEN" in os.environ:
return Scraibe(use_auth_token=os.environ["HF_TOKEN"] )
else:
return Scraibe()