Update Dockerfile

Dockerfile

@@ -7,10 +7,10 @@ ENV HF_HOME=/app/hf_cache
 ENV TRANSFORMERS_CACHE=/app/hf_cache
 ENV HF_TOKEN=${HF_TOKEN}
 ENV PATH=/opt/conda/bin:$PATH
+
 # Install system dependencies
-RUN apt-get update && apt-get install -y \
-    git wget curl unzip ffmpeg libgl1-mesa-glx libglib2.0-0 && \
-    apt-get clean
+RUN apt install libgl1-mesa-glx libgl1-mesa-dri xvfb
+RUN conda install ffmpeg=7 -c conda-forge
 
 # Set up working directory as /app
 WORKDIR /app
@@ -19,16 +19,14 @@ WORKDIR /app
 COPY . /app
 
 # Fix permissions for all subdirectories
-RUN mkdir -p /app/pretrained /app/hf_cache /.cache/gdown && \
+RUN mkdir -p /app/hf_cache /.cache/gdown && \
     chmod -R 777 /app && \
     chmod -R 777 /.cache && \
     chmod -R 777 /root
 
 # Create conda environment and install dependencies
 COPY requirements.txt /app/requirements.txt
-RUN conda create -n epic python=3.10 -y && \
-    conda run -n epic pip install --upgrade pip && \
-    conda run -n epic pip install -r /app/requirements.txt
+RUN pip install -r requirements.txt
 
 RUN chmod -R 777 /app /workspace