Przemyslaw Chachaj commited on
Commit
7a2ad2e
·
1 Parent(s): d5cff7d
Files changed (1) hide show
  1. modules/transcriber.py +13 -0
modules/transcriber.py CHANGED
@@ -1,8 +1,21 @@
1
  import yt_dlp
2
  import os
3
  import uuid
 
4
  from openai import OpenAI
5
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  def transcribe_from_url(url: str) -> str:
7
  """
8
  Stable IG/TikTok/YT reel transcribe
 
1
  import yt_dlp
2
  import os
3
  import uuid
4
+
5
  from openai import OpenAI
6
 
7
+ USE_OPENAI = False # True → OpenAI Whisper, False → Nebius
8
+
9
+ if USE_OPENAI:
10
+ client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
11
+ STT_MODEL = "gpt-4o-transcribe"
12
+ else:
13
+ client = OpenAI(
14
+ base_url="https://api.tokenfactory.nebius.com/v1/",
15
+ api_key=os.environ["NEBIUS_API_KEY"]
16
+ )
17
+ STT_MODEL = "gpt-whisper-1"
18
+
19
  def transcribe_from_url(url: str) -> str:
20
  """
21
  Stable IG/TikTok/YT reel transcribe