feat: Update app for GPU support and modify requirements for torch

app.py

@@ -1,5 +1,6 @@
 import logging
 import tempfile
+import torch
 
 import gradio as gr
 
@@ -18,7 +19,12 @@ logger = logging.getLogger(__name__)
 def transcribe_audio(audio_path):
     """Process audio file and return SRT content and preview text"""
     try:
+        # Check if CUDA is available and set the device
+        device = "cuda" if torch.cuda.is_available() else "cpu"
+        logger.info(f"Using device: {device}")
+
         transcriber = AutoTranscriber(
+            device=device,
             corrector="opencc",
             use_denoiser=False,
             with_punct=False

requirements.txt

@@ -10,4 +10,4 @@ pytest
 pytubefix
 resampy
 torch
-torchaudio
+torchaudio