added logging for cuda

app.py

@@ -1,5 +1,10 @@
 import gradio as gr
 from audio_processing import process_audio, print_results
+import torch
+
+print(f"CUDA available: {torch.cuda.is_available()}")
+print(f"Current device: {torch.cuda.current_device()}")
+print(f"Device name: {torch.cuda.get_device_name()}")
 
 def transcribe_audio(audio_file, translate, model_size):
     language_segments, final_segments = process_audio(audio_file, translate=translate, model_size=model_size)

audio_processing.py

@@ -34,6 +34,7 @@ def process_audio(audio_file, translate=False, model_size="small"):
     
     try:
         device = "cuda" if torch.cuda.is_available() else "cpu"
+        print(f"Using device: {device}")
         compute_type = "float32"
         audio = whisperx.load_audio(audio_file)
         model = whisperx.load_model(model_size, device, compute_type=compute_type)

requirements.txt

@@ -14,4 +14,5 @@ setuptools>=65
 nltk
 python-dotenv
 cdifflib
-pydub
+pydub
+cuda-python