fix huggingface space issue

README.md

@@ -3,7 +3,7 @@ title: Load Balancer
 emoji: 🚀
 colorFrom: yellow
 colorTo: red
-sdk: gradio
+sdk: docker
 sdk_version: 4.36.1
 app_file: app.py
 pinned: false

dockerfile

@@ -0,0 +1,54 @@
+# Base image
+FROM docker.io/library/python:3.10@sha256:428430918ab72109921ef55d5f8bdd3a02a90734fd79ee920146cf385a13c110
+
+# Install additional packages and setup
+RUN apt-get update && apt-get install -y \
+    fakeroot \
+    git \
+    git-lfs \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    cmake \
+    rsync \
+    libgl1-mesa-glx \
+    && rm -rf /var/lib/apt/lists/* \
+    && mv /usr/bin/apt-get /usr/bin/.apt-get \
+    && echo '#!/usr/bin/env sh\nfakeroot /usr/bin/.apt-get $@' > /usr/bin/apt-get \
+    && chmod +x /usr/bin/apt-get \
+    && useradd -m -u 1000 user \
+    && git lfs install
+
+# Set the working directory
+WORKDIR /home/user/app
+
+# Copy requirements file and install dependencies
+COPY requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -r /tmp/requirements.txt
+
+# Freeze the installed packages (optional)
+RUN pip freeze > /tmp/freeze.txt
+
+# Install additional Python packages
+RUN pip install --no-cache-dir \
+    datasets \
+    "huggingface-hub>=0.19" \
+    "hf-transfer>=0.1.4" \
+    "protobuf<4" \
+    "click<8.1" \
+    "pydantic~=1.0" \
+    gradio[oauth]==4.36.1 \
+    "uvicorn>=0.14.0" \
+    spaces
+
+# Copy the application code
+COPY --chown=1000:1000 ./ /home/user/app
+
+# Copy the freeze.txt file
+COPY --chown=1000:1000 /tmp/freeze.txt /tmp/freeze.txt
+
+# Expose the port the app runs on
+EXPOSE 7860
+
+# Set the entry point to run the Flask application
+ENTRYPOINT ["python", "app.py"]