entrypoint.sh

#!/bin/bash

set -e

sleep 5

if [ ! -e "/opt/run/app.py" ]; then
echo 'import gradio as gr
from fastapi import FastAPI

css="""
footer {
    visibility: hidden;
}
"""

_theme = gr.themes.Ocean()

demo = gr.Interface(
    css=css,
    theme=_theme,
    fn=lambda x: x,
    title="3D(Demo)",
    inputs=gr.Model3D(label="Input"),
    outputs=[
        gr.Model3D(clear_color=(0.0, 0.0, 0.0, 0.0), label="Output"),
    ],
    cache_examples=False,
    allow_flagging="never",
)

app = FastAPI()
app = gr.mount_gradio_app(app, demo, path="/")' | tee /opt/run/app.py
fi

echo "========= log: $USER ========="

echo "mamba run pip install --no-cache-dir -r /opt/run/ai-infra.txt" > /usr/bin/ai-infra

if [ -n "$AI_SOCKET" ]; then
    # echo "mamba run hypercorn -b unix:$AI_SOCKET -w 4 /opt/run/app:app" > /usr/bin/ai-infra
    echo "mamba run hypercorn -b unix:$AI_SOCKET /opt/run/app:app" >> /usr/bin/ai-infra
else
    echo "mamba run hypercorn -b '[::]:9100' /opt/run/app:app" >> /usr/bin/ai-infra
fi

exec "$@"