test

app.py

@@ -17,16 +17,13 @@ MODEL = "facebook/wav2vec2-large-960h"
 # MODEL = "patrickvonplaten/wav2vec2-large-960h-lv60-self-4-gram"
 
 
-# is cuda available?
-cuda = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-device = 0 if torch.cuda.is_available() else -1
 speech_recognizer = pipeline(
     task="automatic-speech-recognition",
     model=f"{MODEL}",
     tokenizer=f"{MODEL}",
     framework="pt",
-    device=device,
 )
+speech_recognizer.to("cuda")
 
 videos_out_path = Path("./videos_out")
 videos_out_path.mkdir(parents=True, exist_ok=True)