torch.cuda device automatic detection

app.py

@@ -18,9 +18,12 @@ def asr_transcriber(audio_file):
 
     audio_file_wav = audio_converter(audio_file)
 
-    device_id = "cpu"
-    flash = False
-   
+    # Check for CUDA availability (GPU)
+    if torch.cuda.is_available():
+        device_id = torch.device('cuda')
+    else:
+        device_id = torch.device('cpu')
+
     # Initialize the ASR pipeline
     pipe = pipeline(
         "automatic-speech-recognition",
@@ -28,11 +31,7 @@ def asr_transcriber(audio_file):
         torch_dtype=torch.float16,
         device=device_id
     )
-    if device_id == "mps":
-            torch.mps.empty_cache()
-    elif not flash:
-        pipe.model = pipe.model.to_bettertransformer()
-        
+       
     ts = True
     language = None
     task = "transcribe"