Upload folder using huggingface_hub

Dockerfile

@@ -8,28 +8,30 @@ WORKDIR /app
 RUN apt-get update && apt-get install -y \
     build-essential \
     libgl1-mesa-glx \
+    libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy the requirements file into the container
+# Copy requirements and install
 COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
 
-# Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy everything from the current directory (where Dockerfile is)
+# Copy the rest of the application
 COPY . .
 
-# Create uploads directory
-RUN mkdir -p uploads && chmod 777 uploads
+# Create uploads and set permissions for user 1000
+RUN mkdir -p /app/uploads && \
+    chmod -R 777 /app/uploads && \
+    chown -R 1000:1000 /app
+
+# Swtich to the user with id 1000
+USER 1000
 
-# Create a non-root user and switch to it (Hugging Face Recommended)
-RUN useradd -m -u 1000 user
-USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-# Expose port 7860 (Hugging Face default)
 EXPOSE 7860
 
-# Command to run the application using gunicorn
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "server:app"]
+# Run gunicorn with increased timeout for model loading
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "server:app", "--timeout", "120", "--workers", "1"]
+

requirements.txt

@@ -1,6 +1,7 @@
-tensorflow==2.15.0
+tensorflow-cpu==2.15.0
+numpy==1.26.4
 matplotlib
-numpy
 flask
 Pillow
 gunicorn
+h5py