Restart the Stable Diffusion 2.1

README.md

@@ -4,11 +4,10 @@ emoji: 🔥
 colorFrom: yellow
 colorTo: gray
 sdk: gradio
-sdk_version: 4.44.1
+sdk_version: 3.11.0
 app_file: app.py
 pinned: true
 license: mit
-disable_embedding: true
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -5,12 +5,10 @@ from PIL import Image
 import re
 import os
 import requests
-import uuid
-import base64 
 
 from share_btn import community_icon_html, loading_icon_html, share_js
 
-word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt")
+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
@@ -18,28 +16,20 @@ def infer(prompt, negative, scale):
     global is_gpu_busy
     for filter in word_list:
         if re.search(rf"\b{filter}\b", prompt):
-            print(filter)
-            print(prompt)
             raise gr.Error("Unsafe content found. Please try again with different prompts.")
         
     images = []
     url = os.getenv('JAX_BACKEND_URL')
-    print(url)
     payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
     images_request = requests.post(url, json = payload)
     for image in images_request.json()["images"]:
-        file_path = f"{uuid.uuid4()}.jpg"
-        with open(file_path, "wb") as f:
-            f.write(base64.b64decode(image))
-        images.append(file_path)
+        image_b64 = (f"data:image/jpeg;base64,{image}")
+        images.append(image_b64)
     
-    return images 
+    return images
     
     
 css = """
-        .gradio-container {
-            max-width: 768px !important;
-        }
         .gradio-container {
             font-family: 'IBM Plex Sans', sans-serif;
         }
@@ -57,6 +47,7 @@ css = """
         .container {
             max-width: 730px;
             margin: auto;
+            padding-top: 1.5rem;
         }
         #gallery {
             min-height: 22rem;
@@ -137,9 +128,6 @@ css = """
             margin-top: 10px;
             margin-left: auto;
         }
-        #share-btn-container .styler{
-            background-color: #000000;
-        }
         #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;
         }
@@ -163,7 +151,6 @@ css = """
         #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}
-        button{height: 100%}
 """
 
 block = gr.Blocks(css=css)
@@ -260,28 +247,40 @@ with block:
         """
     )
     with gr.Group():
-        with gr.Row(elem_id="prompt-container"):
-            with gr.Column(scale=3):
-                text = gr.Textbox(
-                    label="Enter your prompt",
-                    show_label=False,
-                    max_lines=1,
-                    placeholder="Enter your prompt",
-                    elem_id="prompt-text-input",
+        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,
                 )
-                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",
-                )
-            with gr.Column(scale=1, min_width=150):
-                btn = gr.Button("Generate image")
 
         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")
@@ -307,9 +306,9 @@ with block:
 
         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], concurrency_limit=80)
-        text.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], concurrency_limit=80)
-        btn.click(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], concurrency_limit=80)
+        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,
@@ -325,7 +324,7 @@ with block:
             None,
             [],
             [],
-            js=share_js,
+            _js=share_js,
         )
         gr.HTML(
             """
@@ -345,5 +344,6 @@ Despite how impressive being able to turn text into image is, beware to the fact
                </div>
                 """
             )
+        
 
-block.queue().launch(max_threads=150, show_error=True)
+block.queue(concurrency_count=80, max_size=100).launch(max_threads=150)