Update app.py

app.py

@@ -13,13 +13,12 @@ lang = "fi"
 
 share = (os.environ.get("SHARE", "False")[0].lower() in "ty1") or None
 auth_token = os.environ.get("AUTH_TOKEN") or True
-device = 0 if torch.cuda.is_available() else "cpu"
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")
 pipe = pipeline(
     task="automatic-speech-recognition",
     model=MODEL_NAME,
     chunk_length_s=30,
-    device=device,
     token=auth_token,
 )
 
@@ -27,6 +26,7 @@ pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(lan
 
 @spaces.GPU(duration=120)
 def transcribe(microphone, file_upload):
+    pipe.to(device)
     warn_output = ""
     if (microphone is not None) and (file_upload is not None):
         warn_output = (