init

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 run -n gradio-user pip install diffusers["torch"] transformers accelerate xformers
+RUN conda run -n gradio-user pip install gradio 
+
+# 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"]

README.md

@@ -1,8 +1,8 @@
 ---
-title: Docker Test17
-emoji: 🐠
-colorFrom: red
-colorTo: blue
+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,16 @@
+#!/bin/bash
+
+CONDA_ENV=$(head -1 /code/environment.yml | cut -d" " -f2)
+eval "$(conda shell.bash hook)"
+conda activate gradio-user
+
+# Kill existing app.py process if running
+# if pgrep -f "python app.py" > /dev/null; then
+#     echo "Stopping existing app.py..."
+#     pkill -f "python app.py"
+#     sleep 2  # Small delay to ensure it stops
+# fi
+
+# Start app.py
+echo "Starting app.py..."
+python app.py