Update app.py

app.py

@@ -16,6 +16,9 @@ tokenizer = Wav2Vec2CTCTokenizer.from_pretrained(model_name, use_auth_token=toke
 model = Wav2Vec2ForCTC.from_pretrained(model_name, use_auth_token=token_key)
 processor = Wav2Vec2ProcessorWithLM.from_pretrained(model_name, use_auth_token=token_key)
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model = model.to(device)
+
 p = pipeline("automatic-speech-recognition", model=model, tokenizer=tokenizer, feature_extractor=processor.feature_extractor, decoder=processor.decoder, use_auth_token=token_key)
 
 #p = pipeline("automatic-speech-recognition", model=model_name, use_auth_token = token_key)