Update Dockerfile

Dockerfile

@@ -1,17 +1,26 @@
 FROM pytorch/pytorch:latest
 
-RUN useradd -m -u 1000 user
-USER user
 # Set working directory
 WORKDIR /app
-ENV PATH="/home/user/.local/bin:$PATH"
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     libgl1-mesa-glx
     
 
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir \
+    torch \
+    torchvision \
+    torchaudio \
+    torchao \
+    git+https://github.com/huggingface/transformers.git \
+    diffusers \
+    Pillow \
+    byaldi \
+    qwen_vl_utils \
+    flash-attn \
+    fastapi \
+    "uvicorn[standard]"
 
 # Install additional required libraries
 RUN pip install byaldi qwen-vl-utils
@@ -19,10 +28,6 @@ RUN pip install byaldi qwen-vl-utils
 # Copy your application code
 COPY app.py .
 
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
-
-COPY --chown=user . /app
 # Expose the port the app runs on
 EXPOSE 8000