added requirements for cuda

app.py

@@ -3,8 +3,7 @@ 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()}")
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 
 def transcribe_audio(audio_file, translate, model_size):
     language_segments, final_segments = process_audio(audio_file, translate=translate, model_size=model_size)

requirements.txt

@@ -7,7 +7,8 @@ pyperclip
 sentencepiece
 gradio
 speechbrain==0.5.16
-torch>=2
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
 torchaudio>=2
 faster-whisper==1.0.0
 setuptools>=65