Update handler.py

handler.py

@@ -7,7 +7,7 @@ from transformers.pipelines.audio_utils import ffmpeg_read
 SAMPLE_RATE=16000
 class EndpointHandler():
     def __init__(self, path=""):
-        #device = "cuda:0" if torch.cuda.is_available() else "cpu"
+        self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
         self.pipe = pipeline(
           "automatic-speech-recognition",
           model="openai/whisper-large",