update

Dockerfile

@@ -0,0 +1,38 @@
+FROM continuumio/anaconda3:main
+
+WORKDIR /code
+COPY ./environment.yml /code/environment.yml
+
+# Create the environment using the environment.yml file
+RUN conda env create -f /code/environment.yml
+
+# install pip packages to the gradio environment
+RUN conda run -n gradio pip install --upgrade pip
+RUN conda run -n gradio diffusers["torch"] transformers accelerate xformers
+RUN conda run -n gradio pip install gradio
+
+# Set the environment variable to use the gradio environment by default
+RUN echo "source activate gradio" > ~/.bashrc
+ENV PATH /opt/conda/envs/gradio/bin:$PATH
+
+# 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"]

README.md

@@ -1,7 +1,7 @@
 ---
-title: Docker Test8
-emoji: 🏃
-colorFrom: purple
+title: Docker Test6
+emoji: 👀
+colorFrom: pink
 colorTo: pink
 sdk: docker
 pinned: false

app.py

@@ -0,0 +1,17 @@
+import gradio as gr
+import torch
+
+
+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()

environment.yml

@@ -0,0 +1,7 @@
+name: gradio                                                                                                                                                                     
+channels:                                                                                                                                                                        
+  - conda-forge       
+  - defaults
+dependencies:
+  - python=3.11
+  - gradio

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 --name "$ENV_NAME" 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