From 018a2bd1cf91f6f8356f4dde3f7a4c626ccf0ee2 Mon Sep 17 00:00:00 2001
From: Tryndaron <aaron.roggenland@gmx.net>
Date: Wed, 3 Apr 2024 16:10:15 +0200
Subject: [PATCH] env hf token

---
 test/test_autotranscript.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/test/test_autotranscript.py b/test/test_autotranscript.py
index b8223c1..9958265 100644
--- a/test/test_autotranscript.py
+++ b/test/test_autotranscript.py
@@ -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()
+