fix docker file

Dockerfile

@@ -37,9 +37,16 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     ffmpeg libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
-ENV HOME=/root \
+
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+
+# Switch to the "user" user
+USER user
+
+ENV HOME=/home/user \
     CUDA_HOME=/usr/local/cuda \
-    PATH=/root/.local/bin:$PATH \
+    PATH=/home/user/.local/bin:$PATH \
     LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
     LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
 
@@ -56,6 +63,6 @@ RUN pip install git+https://github.com/elliottzheng/batch-face.git@master
 RUN pip install --no-cache-dir -r requirements.txt
 ENV PYTHONUNBUFFERED=1
 
-CMD ["gradio", "app.py"]
+CMD ["python", "app.py"]
 
 

app.py

@@ -81,7 +81,7 @@ def extract_audio(video_source, output_dir='./'):
 
     ff = FFmpeg(
         inputs={video_source: None},
-        outputs={output_audio: '-acodec pcm_s24le -ar 48000 -q:a 0 -map a  -y'}
+        outputs={output_audio: '-acodec libmp3lame -ar 44100 -q:a 0 -map a  -y'}
     )
 
     print('ffmpeg command: ', ff.cmd)