Staticaliza commited on
Commit
421ae5f
·
verified ·
1 Parent(s): 0bc0be3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,12 +14,12 @@ print(f"[SYSTEM] | Using {DEVICE} type compute device.")
14
  # Variables
15
  BATCH_SIZE = 8
16
 
17
- pipe = pipeline(task="automatic-speech-recognition", model="openai/whisper-large-v3-turbo", chunk_length_s=30, device=device)
18
 
19
  @spaces.GPU
20
  def transcribe(inputs, task):
21
  if inputs is None: raise gr.Error("Invalid input.")
22
- output = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
23
  return output
24
 
25
  def cloud():
 
14
  # Variables
15
  BATCH_SIZE = 8
16
 
17
+ repo = pipeline(task="automatic-speech-recognition", model="openai/whisper-large-v3-turbo", chunk_length_s=30, device=DEVICE)
18
 
19
  @spaces.GPU
20
  def transcribe(inputs, task):
21
  if inputs is None: raise gr.Error("Invalid input.")
22
+ output = repo(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
23
  return output
24
 
25
  def cloud():