minor updates to match file layout expected by inference runner

tts/StyleTTS2/Dockerfile

@@ -5,6 +5,12 @@ FROM nvidia/cuda:12.2.0-devel-ubuntu20.04
 ENV MY_ENV_VARIABLE=value
 ENV DEBIAN_FRONTEND=noninteractive
 
+RUN mkdir /python_install
+RUN mkdir /model_code
+
+WORKDIR /python_install
+
+
 # Install necessary packages
 RUN apt-get update && apt-get install -y \
     curl \
@@ -12,27 +18,33 @@ RUN apt-get update && apt-get install -y \
     git \
     ffmpeg \
     espeak-ng \
-    sudo \
-    python3.10 \
-    python3-pip
+    sudo
 
 # Install git-lfs
 RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash \
     && apt-get install git-lfs \
     && git lfs install
+# rsync to copy between pods with rich logs
+RUN apt-get install -y rsync
 
-# Create a directory for our project
-WORKDIR /TestingDockerIO
+RUN add-apt-repository ppa:deadsnakes/ppa
 
-# Copy only the necessary files from the host to the container
-COPY StyleTTS2 /TestingDockerIO/StyleTTS2
+RUN apt-get update
+RUN apt-get install -y python3.8 python3.8-venv python3.8-dev python3.8-full python3.8-distutils
+
+COPY requirements.txt .
+# Install Python dependencies
+RUN python3.8 -m venv python \
+  && python/bin/pip  install --upgrade pip wheel setuptools \
+  && . python/bin/activate \
+  && pip install -r requirements.txt
 
 # Change working directory to StyleTTS2
-WORKDIR /TestingDockerIO/StyleTTS2
+WORKDIR /model_code
+
+# Copy only the necessary files from the host to the container
+COPY . .
 
-# Install Python dependencies
-RUN pip3 install --upgrade pip && \
-    pip3 install -r requirements.txt
 
 # Clone the Git LFS repository and move the Models directory
 RUN git lfs clone https://huggingface.co/yl4579/StyleTTS2-LibriTTS && \