Update app.py

app.py

@@ -101,8 +101,8 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 
 
 # Check for CUDA availability
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
-logging.info(f"Using device: {device}")
+# device = "cuda:0" if torch.cuda.is_available() else "cpu"
+# logging.info(f"Using device: {device}")
 
 def download_audio(url, method_choice, proxy_url, proxy_username, proxy_password):
     """
@@ -414,6 +414,9 @@ def get_model_options(pipeline_type):
 # Dictionary to store loaded models
 loaded_models = {}
 
+def get_model_from_cache
+
+
 @spaces.GPU(duration=40)
 def transcribe_audio(audio_input, audio_url, proxy_url, proxy_username, proxy_password, pipeline_type, model_id, download_method, start_time=None, end_time=None, verbose=False):
     """
@@ -435,6 +438,9 @@ def transcribe_audio(audio_input, audio_url, proxy_url, proxy_username, proxy_pa
     Yields:
         Tuple[str, str, str or None]: Metrics and messages, transcription text, path to transcription file.
     """
+    current_device = "cuda:0" if torch.cuda.is_available() else "cpu"
+    device = current_device
+    logging.info(f"Using device: {device}")
     try:
         if verbose:
             logging.getLogger().setLevel(logging.INFO)