Update app.py

app.py

@@ -15,8 +15,8 @@ def format_timestamp(seconds):
 
 def transcribe(audio_file, model_size):
     # Initialize the Whisper model based on the selected model size
-    device = "cpu"  # GPU : cuda  CPU : cpu
-    compute_type = "int8"  # GPU : float16 or int8 - CPU : int8
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    compute_type = "float16" if device == "cuda" else "int8"
     
     model = WhisperModel(model_size, device=device, compute_type=compute_type)