reverted device to cuda

app.py

@@ -39,4 +39,4 @@ iface = gr.Interface(
     title="WhisperX Audio Transcription and Translation"
 )
 
-iface.launch(share=True)
+iface.launch()

audio_processing.py

@@ -35,7 +35,7 @@ def process_audio(audio_file, translate=False, model_size="small"):
     start_time = time.time()
     
     try:
-        device = 0 if torch.cuda.is_available() else "cpu"
+        device = "cuda" if torch.cuda.is_available() else "cpu"
         print(f"Using device: {device}")
         compute_type = "float16"
         audio = whisperx.load_audio(audio_file)

requirements.txt

@@ -14,6 +14,9 @@ faster-whisper==1.0.0
 setuptools>=65
 nltk
 python-dotenv
+ctranslate2
+pytorch
+torchvision
 cdifflib
 pydub
 cuda-python