Update Dockerfile

Dockerfile

@@ -1,19 +1,14 @@
-# Use a lightweight Python image
 FROM python:3.10-slim
 
-# Set working directory
 WORKDIR /app
 
-# Set environment variables for Hugging Face cache and CPU-only execution
 ENV HF_HOME=/tmp/huggingface
 ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
 ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
-# Disable GPU detection
 ENV CUDA_VISIBLE_DEVICES=""
 
-
 RUN apt-get update && apt-get install -y \
     build-essential \
     libgl1-mesa-glx \
@@ -25,25 +20,25 @@ RUN apt-get update && apt-get install -y \
     libopenblas-dev \
     liblapack-dev \
     git \
+    libopencv-dev \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-  
-
-# Create necessary directories with appropriate permissions
 RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggingface/datasets \
     && chmod -R 777 /tmp/uploads /tmp/results /tmp/huggingface
 
-# Copy requirements file and install dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir -U pip && \
-    pip install --no-cache-dir -r requirements.txt
+RUN python -m venv /venv && \
+    /venv/bin/pip install --no-cache-dir -U pip && \
+    /venv/bin/pip install --no-cache-dir -r requirements.txt
+
+RUN git clone https://github.com/baaivision/LGM.git /app/lgm && \
+    /venv/bin/pip install --no-cache-dir /app/lgm
 
-# Copy application code
 COPY app.py .
 
-# Expose the port the app runs on (standard for Hugging Face Spaces)
+ENV PATH="/venv/bin:$PATH"
+
 EXPOSE 7860
 
-# Command to run the application
 CMD ["python", "app.py"]