Update app.py

app.py

@@ -14,8 +14,7 @@ LOCAL_DIR.mkdir(exist_ok=True)
 SAVE_DIR = LOCAL_DIR / "transcripts"
 SAVE_DIR.mkdir(exist_ok=True)
 LANGS = [lang.capitalize() for lang in list(LANGUAGES.values())]
-# DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
-DEVICE = 'cuda'
+DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 loaded_model = whisper.load_model("medium", DEVICE)
 current_size = "None"
 
@@ -34,6 +33,7 @@ def get_transcript(audio_path, task_selection:str, language:str, max_line_width=
         "max_line_width": max_line_width if max_line_width > 0 else None,
         "max_line_count": max_line_count if max_line_count > 0 else None,
         "max_words_per_line": max_words_per_line if max_words_per_line > 0 else None
+
     }
 
     options = dict(task=task_selection.lower(), best_of=5, language=TO_LANGUAGE_CODE[language.lower()])