From 5be7068bf5890aceb1778bbee24a49b8b761c71c Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Sun, 28 Jan 2024 11:40:10 -0700
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 9896245..be1b8a1 100755
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,8 +1,9 @@
-FROM ubuntu:latest
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
 WORKDIR /subgen
 
 RUN apt-get update && apt-get -y install python3 python3-pip ffmpeg
+RUN pip3-install numpy stable-ts fastapi requests faster-whisper uvicorn python-multipart whisper
 
 ENV PYTHONUNBUFFERED 1