diff --git a/back/requirements.txt b/back/requirements.txt
index 2560f88..9a357c2 100644
--- a/back/requirements.txt
+++ b/back/requirements.txt
@@ -1,7 +1,6 @@
 gputil
 fastapi
-transformers
-torch
+transformers==4.37.2
 uvicorn
 intervaltree
 filemagic
diff --git a/back/wp.py b/back/wp.py
index b4b7b8d..c281c4f 100644
--- a/back/wp.py
+++ b/back/wp.py
@@ -23,7 +23,7 @@
     model_kwargs={"use_flash_attention_2": True},
 )
 
-diarize_pipe = Pipeline.from_pretrained("pyannote/speaker-diarization@2.1.1",
+diarize_pipe = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1",
                                         use_auth_token="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")
 diarize_pipe.to(device)