Martí Umbert commited on
Commit
bdcec79
·
1 Parent(s): 5e94898

Dockerfile: to test diarization in HF space

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -24,7 +24,7 @@ RUN apt-get update && \
24
  git && \
25
  rm -rf /var/lib/apt/lists/*
26
 
27
- RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
28
 
29
  COPY . .
30
 
@@ -79,4 +79,4 @@ EXPOSE 8000
79
  ENTRYPOINT ["whisperlivekit-server", "--host", "0.0.0.0"]
80
 
81
  # Default args
82
- CMD ["--model", "tiny.en"]
 
24
  git && \
25
  rm -rf /var/lib/apt/lists/*
26
 
27
+ RUN pip install diart torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
28
 
29
  COPY . .
30
 
 
79
  ENTRYPOINT ["whisperlivekit-server", "--host", "0.0.0.0"]
80
 
81
  # Default args
82
+ CMD ["--model", "tiny.en", "diart"]