update

Dockerfile

@@ -0,0 +1,28 @@
+FROM continuumio/anaconda3:main
+
+# WORKDIR /code
+
+# Create the environment using the environment.yml file
+RUN conda create -n gradio python=3.11 -y
+
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+# Switch to the "user" user
+USER user
+# Set home to the user's home directory
+ENV HOME=/home/user \
+    PYTHONPATH=$HOME/app \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+
+CMD ["./run.sh"]

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+
+
+def update(name):
+    return f"Welcome to Gradio, {name}!"
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("Start typing below and then click **Run** to see the output.")
+    with gr.Row():
+        inp = gr.Textbox(placeholder="What is your name?")
+        out = gr.Textbox()
+    btn = gr.Button("Run")
+    btn.click(fn=update, inputs=inp, outputs=out)
+
+demo.launch()

run.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+set -e  # stop on error
+
+eval "$(conda shell.bash hook)"
+
+ENV_NAME="matgen-plus"
+PYTHON_VER="3.11"
+
+# Create env and install packages only if it doesn't exist
+if ! conda env list | grep -q "^${ENV_NAME}\s"; then
+    echo "Creating conda environment: $ENV_NAME"
+    conda create -y -n "$ENV_NAME" python="$PYTHON_VER"
+    conda activate "$ENV_NAME"
+
+    echo "Installing packages in $ENV_NAME..."
+
+    # pip installs
+    pip install diffusers["torch"] transformers accelerate xformers
+    pip install gradio controlnet-aux
+    pip install trimesh xatlas scikit-learn opencv-python omegaconf
+
+    # conda installs
+    # conda install -y pytorch3d -c pytorch -c conda-forge
+    # conda install -y -c conda-forge open-clip-torch pytorch-lightning
+
+    python app.py
+
+else
+    echo "Conda environment '$ENV_NAME' already exists."
+    conda activate "$ENV_NAME"
+fi
+
+# Run the app
+# python app.py