Update app.py

app.py

@@ -42,6 +42,9 @@ SAMPLE_RATE = 16_000
 dia_model = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=os.environ['TOKEN'])
 vad_model = Pipeline.from_pretrained("pyannote/voice-activity-detection", use_auth_token=os.environ['TOKEN'])
 
+dia_model = dia_model.to(torch.device('cuda'))
+vad_model = vad_model.to(torch.device('cuda'))
+
 import torch
 from transformers import pipeline
 from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq