update

Dockerfile

@@ -0,0 +1,50 @@
+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
+# when adjusting the dockerfile on huggingface:
+# - if the dockerfile is successfully compileds, a new space need to be initialized and push the changes accordingly
+# - otherwise, you can commit to the failed build dockerfile for debugging
+RUN conda run -n gradio pip install --upgrade pip
+# RUN conda run -n gradio pip install diffusers["torch"] transformers accelerate xformers
+# RUN conda run -n gradio pip install gradio 
+# RUN conda run -n gradio pip install controlnet-aux
+# RUN conda install -n gradio pytorch3d -c pytorch3d -c conda-forge
+# RUN conda install -n gradio -c conda-forge open-clip-torch pytorch-lightning
+# RUN conda run -n gradio pip install trimesh xatlas scikit-learn opencv-python omegaconf
+
+# 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
+
+RUN conda create -n gradio-user python=3.11
+RUN conda run -n gradio-user pip install --upgrade pip
+# RUN conda install -n gradio-user pytorch3d=0.7.7 -c pytorch3d -c conda-forge
+# RUN conda install -n gradio-user -c conda-forge open-clip-torch pytorch-lightning
+RUN conda run -n gradio-user pip install diffusers transformers accelerate xformers controlnet-aux gradio trimesh xatlas scikit-learn opencv-python omegaconf
+
+# Set home to the user's home directory
+ENV HOME=/home/user \
+    PYTHONPATH=$HOME/app \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    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,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,14 @@
+#!/bin/bash
+
+CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
+eval "$(conda shell.bash hook)"
+conda activate gradio-user
+
+conda install -n gradio-user pytorch3d=0.7.7 -c pytorch3d -c conda-forge
+conda install -n gradio-user -c conda-forge open-clip-torch pytorch-lightning
+
+# Start app.py
+echo "Starting app.py..."
+export OMP_NUM_THREADS=4 # default is a wrong value: 7500m
+python -c "import torch; x=torch.rand(1, device='cuda'); print(x, x.device.type)"
+python app.py