Update handler.py
Browse files- 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,
|