pipe device

app.py

@@ -80,6 +80,7 @@ def transcribe(youtube_url, audio_file, whisper_model, gemini_api_key, gemini_pr
             chunk_length_s=30,
             device=device,
         )
+        pipe.model = pipe.model.to(device)
 
         if youtube_url:
             progress(0.1, desc="Extracting YouTube ID")