yavorr commited on
Commit
22be480
·
1 Parent(s): 48324cb

Make whisper client init lazy as we don't need it for TTS and VC

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -5,10 +5,17 @@ from openai import OpenAI
5
 
6
  from playdiffusion import PlayDiffusion, InpaintInput, TTSInput, RVCInput
7
 
8
- whisper_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
9
  inpainter = PlayDiffusion()
 
 
 
 
 
 
 
10
 
11
  def run_asr(audio):
 
12
  audio_file = open(audio, "rb")
13
  transcript = whisper_client.audio.transcriptions.create(
14
  file=audio_file,
 
5
 
6
  from playdiffusion import PlayDiffusion, InpaintInput, TTSInput, RVCInput
7
 
 
8
  inpainter = PlayDiffusion()
9
+ _whisper_client = None
10
+
11
+ def get_whisper_client():
12
+ global _whisper_client
13
+ if _whisper_client is None:
14
+ _whisper_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
15
+ return _whisper_client
16
 
17
  def run_asr(audio):
18
+ whisper_client = get_whisper_client()
19
  audio_file = open(audio, "rb")
20
  transcript = whisper_client.audio.transcriptions.create(
21
  file=audio_file,