avfranco commited on
Commit
554a124
1 Parent(s): 821d391

dtype=float32

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,12 +23,12 @@ def asr_transcriber(audio_file):
23
  device_id = torch.device('cuda')
24
  else:
25
  device_id = torch.device('cpu')
26
-
27
  # Initialize the ASR pipeline
28
  pipe = pipeline(
29
  "automatic-speech-recognition",
30
  model="openai/whisper-large-v3",
31
- torch_dtype=torch.float16,
32
  device=device_id
33
  )
34
 
 
23
  device_id = torch.device('cuda')
24
  else:
25
  device_id = torch.device('cpu')
26
+
27
  # Initialize the ASR pipeline
28
  pipe = pipeline(
29
  "automatic-speech-recognition",
30
  model="openai/whisper-large-v3",
31
+ torch_dtype=torch.float32,
32
  device=device_id
33
  )
34