S-Fry commited on
Commit
d3d7605
1 Parent(s): 07e34c4

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -6,8 +6,8 @@ from datasets import load_dataset
6
  SAMPLE_RATE=16000
7
  class EndpointHandler():
8
  def __init__(self, path=""):
9
- device = "cuda:0" if torch.cuda.is_available() else "cpu"
10
- pipe = pipeline(
11
  "automatic-speech-recognition",
12
  model="openai/whisper-large",
13
  chunk_length_s=30,
 
6
  SAMPLE_RATE=16000
7
  class EndpointHandler():
8
  def __init__(self, path=""):
9
+ #device = "cuda:0" if torch.cuda.is_available() else "cpu"
10
+ self.pipe = pipeline(
11
  "automatic-speech-recognition",
12
  model="openai/whisper-large",
13
  chunk_length_s=30,