fixes to yaml after guy review

mlrun · Dec 11, 2023 · 40e84f3 · 40e84f3
1 parent e09e878
commit 40e84f3
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/speech_diarization/item.yaml b/speech_diarization/item.yaml
@@ -1,7 +1,8 @@
 apiVersion: v1
 categories:
-- Utilities
-- Machine Learning
+  - Deep Learning
+  - Huggingface
+  - Audio
 description: speech diarization of audio files
 doc: ''
 example: speech_diarization.ipynb
@@ -18,8 +19,12 @@ platformVersion: 3.5.3
 spec:
   filename: speech_diarization.py
   handler: diarize
-  image: mlrun/mlrun
+  image: mlrun/mlrun-gpu
   kind: job
-  requirements: []
+  requirements:
+    -pyannote.audio
+    -pyannote.core
+    -torchaudio
+    -tqdm
 url: ''
 version: 1.0.0