Update app.py

app.py

@@ -35,7 +35,8 @@ def format_and_group_timestamps(chunks, interval=5.0):
 def transcribe_audio(input_file, language, chunk_length_s=10, stride_length_s=(4, 2), return_timestamps="word"):
     target_lang_options = {"English": "eng", "Luganda": "lug", "Acholi": "ach", "Runyankole": "nyn", "Lugbara": "lgg"}
     target_lang_code = target_lang_options[language]
-
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    
     if target_lang_code == "eng":
         # Use Whisper for English
         model = whisper.load_model("small")
@@ -44,7 +45,6 @@ def transcribe_audio(input_file, language, chunk_length_s=10, stride_length_s=(4
         return result["text"]
     else:
         # Use specified model for other languages
-        device = "cuda" if torch.cuda.is_available() else "cpu"
         model_id = "Sunbird/sunbird-mms"
 
         auth_token = os.environ.get("HF_TOKEN")