Update handler.py

handler.py

@@ -9,10 +9,10 @@ class EndpointHandler():
         torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
         model_id = "openai/whisper-large-v3"
 
-        #model = AutoModelForSpeechSeq2Seq.from_pretrained(
-        #    model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
-        #)
-        #model.to(device)
+        model = AutoModelForSpeechSeq2Seq.from_pretrained(
+            model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
+        )
+        model.to(device)
 
         processor = AutoProcessor.from_pretrained(model_id)
         #processor = WhisperProcessor.from_pretrained(model_id)