commit 806605c16fad44009cc9cc979a6b6f2e6563bbb2 Author: jianjiang Date: Tue Apr 18 14:13:57 2023 +0800 new file: share_btn.py diff --git a/.gitea/workflows/build.yaml b/.gitea/workflows/build.yaml new file mode 100644 index 0000000..03be22b --- /dev/null +++ b/.gitea/workflows/build.yaml @@ -0,0 +1,47 @@ +name: Build +run-name: ${{ github.actor }} is upgrade release 🚀 +on: [push] +env: + REPOSITORY: ${{ github.repository }} + COMMIT_ID: ${{ github.sha }} +jobs: + Build-Deploy-Actions: + runs-on: ubuntu-latest + steps: + - run: echo "🎉 The job was automatically triggered by a ${{ github.event_name }} event." + - run: echo "🐧 This job is now running on a ${{ runner.os }} server hosted by Gitea!" + - run: echo "🔎 The name of your branch is ${{ github.ref }} and your repository is ${{ github.repository }}." + - name: Check out repository code + uses: actions/checkout@v3 + - + name: Setup Git LFS + run: | + git lfs install + git lfs fetch + git lfs checkout + - name: List files in the repository + run: | + ls ${{ github.workspace }} + - + name: Docker Image Info + id: image-info + run: | + echo "::set-output name=image_name::$(echo $REPOSITORY | tr '[:upper:]' '[:lower:]')" + echo "::set-output name=image_tag::${COMMIT_ID:0:10}" + - + name: Login to Docker Hub + uses: docker/login-action@v2 + with: + registry: artifacts.iflytek.com + username: ${{ secrets.DOCKERHUB_USERNAME }} + password: ${{ secrets.DOCKERHUB_TOKEN }} + - name: Set up Docker Buildx + uses: docker/setup-buildx-action@v2 + - + name: Build and push + run: | + docker version + docker buildx build -t artifacts.iflytek.com/docker-private/atp/${{ steps.image-info.outputs.image_name }}:${{ steps.image-info.outputs.image_tag }} . --file ${{ github.workspace }}/Dockerfile --load + docker push artifacts.iflytek.com/docker-private/atp/${{ steps.image-info.outputs.image_name }}:${{ steps.image-info.outputs.image_tag }} + docker rmi artifacts.iflytek.com/docker-private/atp/${{ steps.image-info.outputs.image_name }}:${{ steps.image-info.outputs.image_tag }} + - run: echo "🍏 This job's status is ${{ job.status }}." diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..c39b05c --- /dev/null +++ b/Dockerfile @@ -0,0 +1,27 @@ +# please visit https://github.com/xfyun/aiges/releases to get stable and suitable iamges. + +FROM iflyopensource/aiges-gpu:11.2-1.17-3.9.13-ubuntu1804-v3.3.2 +RUN sed -i 's@//.*archive.ubuntu.com@//mirrors.ustc.edu.cn@g' /etc/apt/sources.list +RUN sed -i 's/security.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list + +RUN --mount=target=/root/packages.txt,source=packages.txt apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/* + +WORKDIR /home/user/app +RUN useradd -m -u 1000 user +RUN chown -R 1000.1000 /home/user + + +RUN pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/ + + +RUN pip install --no-cache-dir pip==22.3.1 +RUN --mount=target=requirements.txt,source=requirements.txt pip install --no-cache-dir -r requirements.txt + +COPY --chown=1000 ./ /home/user/app + +RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash +RUN apt-get install git-lfs +RUN git lfs install +RUN git clone http://172.16.59.16:3000/microsoft/table-transformer-detection.git + +CMD ["python3", "app.py"] diff --git a/app.py b/app.py new file mode 100644 index 0000000..2b67184 --- /dev/null +++ b/app.py @@ -0,0 +1,349 @@ +import gradio as gr +from datasets import load_dataset +from PIL import Image + +import re +import os +import requests + +from share_btn import community_icon_html, loading_icon_html, share_js + +word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True) +word_list = word_list_dataset["train"]['text'] + +is_gpu_busy = False +def infer(prompt, negative, scale): + global is_gpu_busy + for filter in word_list: + if re.search(rf"\b{filter}\b", prompt): + raise gr.Error("Unsafe content found. Please try again with different prompts.") + + images = [] + url = os.getenv('JAX_BACKEND_URL') + payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale} + images_request = requests.post(url, json = payload) + for image in images_request.json()["images"]: + image_b64 = (f"data:image/jpeg;base64,{image}") + images.append(image_b64) + + return images + + +css = """ + .gradio-container { + font-family: 'IBM Plex Sans', sans-serif; + } + .gr-button { + color: white; + border-color: black; + background: black; + } + input[type='range'] { + accent-color: black; + } + .dark input[type='range'] { + accent-color: #dfdfdf; + } + .container { + max-width: 730px; + margin: auto; + padding-top: 1.5rem; + } + #gallery { + min-height: 22rem; + margin-bottom: 15px; + margin-left: auto; + margin-right: auto; + border-bottom-right-radius: .5rem !important; + border-bottom-left-radius: .5rem !important; + } + #gallery>div>.h-full { + min-height: 20rem; + } + .details:hover { + text-decoration: underline; + } + .gr-button { + white-space: nowrap; + } + .gr-button:focus { + border-color: rgb(147 197 253 / var(--tw-border-opacity)); + outline: none; + box-shadow: var(--tw-ring-offset-shadow), var(--tw-ring-shadow), var(--tw-shadow, 0 0 #0000); + --tw-border-opacity: 1; + --tw-ring-offset-shadow: var(--tw-ring-inset) 0 0 0 var(--tw-ring-offset-width) var(--tw-ring-offset-color); + --tw-ring-shadow: var(--tw-ring-inset) 0 0 0 calc(3px var(--tw-ring-offset-width)) var(--tw-ring-color); + --tw-ring-color: rgb(191 219 254 / var(--tw-ring-opacity)); + --tw-ring-opacity: .5; + } + #advanced-btn { + font-size: .7rem !important; + line-height: 19px; + margin-top: 12px; + margin-bottom: 12px; + padding: 2px 8px; + border-radius: 14px !important; + } + #advanced-options { + display: none; + margin-bottom: 20px; + } + .footer { + margin-bottom: 45px; + margin-top: 35px; + text-align: center; + border-bottom: 1px solid #e5e5e5; + } + .footer>p { + font-size: .8rem; + display: inline-block; + padding: 0 10px; + transform: translateY(10px); + background: white; + } + .dark .footer { + border-color: #303030; + } + .dark .footer>p { + background: #0b0f19; + } + .acknowledgments h4{ + margin: 1.25em 0 .25em 0; + font-weight: bold; + font-size: 115%; + } + .animate-spin { + animation: spin 1s linear infinite; + } + @keyframes spin { + from { + transform: rotate(0deg); + } + to { + transform: rotate(360deg); + } + } + #share-btn-container { + display: flex; padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; width: 13rem; + margin-top: 10px; + margin-left: auto; + } + #share-btn { + all: initial; color: #ffffff;font-weight: 600; cursor:pointer; font-family: 'IBM Plex Sans', sans-serif; margin-left: 0.5rem !important; padding-top: 0.25rem !important; padding-bottom: 0.25rem !important;right:0; + } + #share-btn * { + all: unset; + } + #share-btn-container div:nth-child(-n+2){ + width: auto !important; + min-height: 0px !important; + } + #share-btn-container .wrap { + display: none !important; + } + + .gr-form{ + flex: 1 1 50%; border-top-right-radius: 0; border-bottom-right-radius: 0; + } + #prompt-container{ + gap: 0; + } + #prompt-text-input, #negative-prompt-text-input{padding: .45rem 0.625rem} + #component-16{border-top-width: 1px!important;margin-top: 1em} + .image_duplication{position: absolute; width: 100px; left: 50px} +""" + +block = gr.Blocks(css=css) + +examples = [ + [ + 'A high tech solarpunk utopia in the Amazon rainforest', + 'low quality', + 9 + ], + [ + 'A pikachu fine dining with a view to the Eiffel Tower', + 'low quality', + 9 + ], + [ + 'A mecha robot in a favela in expressionist style', + 'low quality, 3d, photorealistic', + 9 + ], + [ + 'an insect robot preparing a delicious meal', + 'low quality, illustration', + 9 + ], + [ + "A small cabin on top of a snowy mountain in the style of Disney, artstation", + 'low quality, ugly', + 9 + ], +] + + +with block: + gr.HTML( + """ +
+
+ + + + + + + + + + + + + + + + + + + + + + + + + + + +

+ Stable Diffusion 2.1 Demo +

+
+

+ Stable Diffusion 2.1 is the latest text-to-image model from StabilityAI. Access Stable Diffusion 1 Space here
For faster generation and API + access you can try + DreamStudio Beta. +

+
+ """ + ) + with gr.Group(): + with gr.Box(): + with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True): + with gr.Column(): + text = gr.Textbox( + label="Enter your prompt", + show_label=False, + max_lines=1, + placeholder="Enter your prompt", + elem_id="prompt-text-input", + ).style( + border=(True, False, True, True), + rounded=(True, False, False, True), + container=False, + ) + negative = gr.Textbox( + label="Enter your negative prompt", + show_label=False, + max_lines=1, + placeholder="Enter a negative prompt", + elem_id="negative-prompt-text-input", + ).style( + border=(True, False, True, True), + rounded=(True, False, False, True), + container=False, + ) + btn = gr.Button("Generate image").style( + margin=False, + rounded=(False, True, True, False), + full_width=False, + ) + + gallery = gr.Gallery( + label="Generated images", show_label=False, elem_id="gallery" + ).style(grid=[2], height="auto") + + with gr.Group(elem_id="container-advanced-btns"): + #advanced_button = gr.Button("Advanced options", elem_id="advanced-btn") + with gr.Group(elem_id="share-btn-container"): + community_icon = gr.HTML(community_icon_html) + loading_icon = gr.HTML(loading_icon_html) + share_button = gr.Button("Share to community", elem_id="share-btn") + + with gr.Accordion("Advanced settings", open=False): + # gr.Markdown("Advanced settings are temporarily unavailable") + # samples = gr.Slider(label="Images", minimum=1, maximum=4, value=4, step=1) + # steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=45, step=1) + guidance_scale = gr.Slider( + label="Guidance Scale", minimum=0, maximum=50, value=9, step=0.1 + ) + # seed = gr.Slider( + # label="Seed", + # minimum=0, + # maximum=2147483647, + # step=1, + # randomize=True, + # ) + + ex = gr.Examples(examples=examples, fn=infer, inputs=[text, negative, guidance_scale], outputs=[gallery, community_icon, loading_icon, share_button], cache_examples=False) + ex.dataset.headers = [""] + negative.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False) + text.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False) + btn.click(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False) + + #advanced_button.click( + # None, + # [], + # text, + # _js=""" + # () => { + # const options = document.querySelector("body > gradio-app").querySelector("#advanced-options"); + # options.style.display = ["none", ""].includes(options.style.display) ? "flex" : "none"; + # }""", + #) + share_button.click( + None, + [], + [], + _js=share_js, + ) + gr.HTML( + """ +
+

Model by StabilityAI - backend running JAX on TPUs due to generous support of Google TRC program - Gradio Demo by 🤗 Hugging Face +

+
+ """ + ) + with gr.Accordion(label="License", open=False): + gr.HTML( + """
+

LICENSE

+The model is licensed with a CreativeML OpenRAIL++ license. The authors claim no rights on the outputs you generate, you are free to use them and are accountable for their use which must not go against the provisions set in this license. The license forbids you from sharing any content that violates any laws, produce any harm to a person, disseminate any personal information that would be meant for harm, spread misinformation and target vulnerable groups. For the full list of restrictions please read the license

+

Biases and content acknowledgment

+Despite how impressive being able to turn text into image is, beware to the fact that this model may output content that reinforces or exacerbates societal biases, as well as realistic faces, pornography and violence. The model was trained on the LAION-5B dataset, which scraped non-curated image-text-pairs from the internet (the exception being the removal of illegal content) and is meant for research purposes. You can read more in the model card

+
+ """ + ) + + +block.queue(concurrency_count=80, max_size=100).launch(max_threads=150) diff --git a/packages.txt b/packages.txt new file mode 100644 index 0000000..f027e0d --- /dev/null +++ b/packages.txt @@ -0,0 +1 @@ +vim diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..9bef651 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,3 @@ +python-dotenv +gradio +transformers diff --git a/share_btn.py b/share_btn.py new file mode 100644 index 0000000..a85c7c3 --- /dev/null +++ b/share_btn.py @@ -0,0 +1,60 @@ +community_icon_html = """""" + +loading_icon_html = """""" + +share_js = """async () => { + async function uploadFile(file){ + const UPLOAD_URL = 'https://huggingface.co/uploads'; + const response = await fetch(UPLOAD_URL, { + method: 'POST', + headers: { + 'Content-Type': file.type, + 'X-Requested-With': 'XMLHttpRequest', + }, + body: file, /// <- File inherits from Blob + }); + const url = await response.text(); + return url; + } + const gradioEl = document.querySelector('body > gradio-app'); + const imgEls = gradioEl.querySelectorAll('#gallery img'); + const promptTxt = gradioEl.querySelector('#prompt-text-input input').value; + const shareBtnEl = gradioEl.querySelector('#share-btn'); + const shareIconEl = gradioEl.querySelector('#share-btn-share-icon'); + const loadingIconEl = gradioEl.querySelector('#share-btn-loading-icon'); + if(!imgEls.length){ + return; + }; + shareBtnEl.style.pointerEvents = 'none'; + shareIconEl.style.display = 'none'; + loadingIconEl.style.removeProperty('display'); + const files = await Promise.all( + [...imgEls].map(async (imgEl) => { + const res = await fetch(imgEl.src); + const blob = await res.blob(); + const imgId = Date.now() % 200; + const fileName = `diffuse-the-rest-${{imgId}}.jpg`; + return new File([blob], fileName, { type: 'image/jpeg' }); + }) + ); + const urls = await Promise.all(files.map((f) => uploadFile(f))); + const htmlImgs = urls.map(url => ``); + const descriptionMd = `
+${htmlImgs.join(`\n`)} +
`; + const params = new URLSearchParams({ + title: promptTxt, + description: descriptionMd, + }); + const paramsStr = params.toString(); + window.open(`https://huggingface.co/spaces/stabilityai/stable-diffusion/discussions/new?${paramsStr}`, '_blank'); + shareBtnEl.style.removeProperty('pointer-events'); + shareIconEl.style.removeProperty('display'); + loadingIconEl.style.display = 'none'; +}"""