Update Dockerfile

Dockerfile

@@ -1,16 +1,43 @@
 FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
-SHELL ["/bin/bash","-lc"]
 
-ENV HF_HOME=/app/hf_cache TRANSFORMERS_CACHE=/app/hf_cache DISPLAY=:99
+SHELL ["/bin/bash", "-c"]
 
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    git git-lfs ffmpeg libgl1 libgl1-mesa-dri libglib2.0-0 xvfb \
- && rm -rf /var/lib/apt/lists/*
+# Environment variables for Hugging Face cache
+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
 
+# Set up working directory as /app
 WORKDIR /app
-COPY requirements.txt .
-RUN python -m pip install --upgrade pip && pip install -r requirements.txt
 
-COPY . .
+# Copy project into /app
+COPY . /app
+
+# Fix permissions for all subdirectories
+RUN mkdir -p /app/pretrained /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 realcam python=3.10 -y && \
+    conda run -n realcam pip install --upgrade pip && \
+    conda run -n realcam pip install -r /app/requirements.txt
+
+RUN chmod -R 777 /app /workspace
+
+# # List contents (for debug)
+RUN ls -la /app
+RUN pip install gradio
+
+# Expose Gradio default port
 EXPOSE 7860
-CMD xvfb-run -s "-screen 0 1280x720x24" python finetune/gradio_app.py --server_name 0.0.0.0 --server_port 7860
+
+# Start the Gradio app
+CMD ["conda", "run", "--no-capture-output", "-n", "realcam", "python", "finetune/gradio_app.py"]