b0ss fix b0ss

Dockerfile

@@ -0,0 +1,50 @@
+FROM python:3.10
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    git-lfs \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    cmake \
+    rsync \
+    libgl1 \
+    && rm -rf /var/lib/apt/lists/* \
+    && git lfs install
+
+WORKDIR /app
+
+# Pin pip to a version that can handle omegaconf's invalid metadata
+RUN pip install --no-cache-dir "pip<24.1"
+
+# Install HuggingFace dependencies
+RUN pip install --no-cache-dir \
+    datasets \
+    "huggingface-hub>=0.30" \
+    "hf-transfer>=0.1.4" \
+    "protobuf<4" \
+    "click<8.1" \
+    "pydantic~=1.0"
+
+# Copy and install requirements
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application
+COPY . .
+
+# Create a non-root user for HuggingFace Spaces
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+COPY --chown=user . $HOME/app
+
+# Expose the Streamlit port
+EXPOSE 7860
+
+# Run the Streamlit app
+CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]

requirements.txt

@@ -1,4 +1,3 @@
-pip<24.1
 torch~=2.0.1
 onnxruntime~=1.15.1
 onnxruntime-gpu~=1.15.1