Manjushri commited on
Commit
c2c12af
·
1 Parent(s): 3a2e636

Update app.py

Browse files

Fixed arg error, added nsfw warning

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,7 +23,7 @@ else:
23
 
24
  def genie (prompt, negative_prompt, height, width, scale, steps, seed):
25
  generator = torch.Generator(device=device).manual_seed(seed)
26
- int_image = pipe(prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator, width=768, height=768, output_type="latent").images
27
  image = refiner(prompt=prompt, negative_prompt=negative_prompt, image=int_image).images[0]
28
  return image
29
 
@@ -36,5 +36,5 @@ gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generat
36
  gr.Slider(minimum=1, step=1, maximum=999999999999999999, randomize=True)],
37
  outputs='image',
38
  title="Stable Diffusion XL 1.0 CPU or GPU",
39
- description="SDXL 1.0 CPU or GPU. Currently running on CPU. <br><br><b>WARNING:</b> Extremely Slow. 65s/Iteration. Expect 25-50mins an image for 25-50 iterations respectively.",
40
  article = "Code Monkey: <a href=\"https://huggingface.co/Manjushri\">Manjushri</a>").launch(debug=True, max_threads=80)
 
23
 
24
  def genie (prompt, negative_prompt, height, width, scale, steps, seed):
25
  generator = torch.Generator(device=device).manual_seed(seed)
26
+ int_image = pipe(prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator, output_type="latent").images
27
  image = refiner(prompt=prompt, negative_prompt=negative_prompt, image=int_image).images[0]
28
  return image
29
 
 
36
  gr.Slider(minimum=1, step=1, maximum=999999999999999999, randomize=True)],
37
  outputs='image',
38
  title="Stable Diffusion XL 1.0 CPU or GPU",
39
+ description="SDXL 1.0 CPU or GPU. Currently running on CPU. <br><br><b>WARNING:</b> Extremely Slow. 65s/Iteration. Expect 25-50mins an image for 25-50 iterations respectively. This model is capable of producing NSFW (Softcore) images.",
40
  article = "Code Monkey: <a href=\"https://huggingface.co/Manjushri\">Manjushri</a>").launch(debug=True, max_threads=80)