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)