good app

app.py

@@ -1,69 +1,80 @@
 # pip install transformers gradio scipy ftfy "ipywidgets>=7,<8" datasets diffusers
+import random
 
 import gradio as gr
 import torch
 from torch import autocast
-from diffusers import StableDiffusionPipeline
+from diffusers.pipelines.stable_diffusion import StableDiffusionPipeline
 
 model_id = "hakurei/waifu-diffusion"
-device = "cpu"
-
-# pipe = StableDiffusionPipeline.from_pretrained(model_id,
-#                                                resume_download=True,  # 模型文件断点续传
-#                                                torch_dtype=torch.float16,
-#                                                revision='fp16')
-# pipe = pipe.to(device)
-
-block = gr.Blocks(css=".container { max-width: 800px; margin: auto; }")
-
-num_samples = 2
-
-
-def infer(prompt):
-    with autocast("cuda"):
-        images = pipe([prompt] * num_samples,
-                      hight=111,
-                      width=100,
-
-                      guidance_scale=7.5)["sample"]
-
-    return images
-
-
-with block as demo:
-    gr.Markdown("<h1><center>Waifu Diffusion</center></h1>")
-    gr.Markdown(
-        "waifu-diffusion is a latent text-to-image diffusion model that has been conditioned on high-quality anime images through fine-tuning."
-    )
-    with gr.Group():
-        with gr.Box():
-            with gr.Row().style(mobile_collapse=False, equal_height=True):
-                text = gr.Textbox(
-                    label="Enter your prompt", show_label=False, max_lines=1
-                ).style(
-                    border=(True, False, True, True),
-                    rounded=(True, False, False, True),
-                    container=False,
-                )
-                btn = gr.Button("Run").style(
-                    margin=False,
-                    rounded=(False, True, True, False),
-                )
-
-        gallery = gr.Gallery(label="Generated images", show_label=False).style(
-            grid=[2], height="auto"
-        )
-        text.submit(infer, inputs=[text], outputs=gallery)
-        btn.click(infer, inputs=[text], outputs=gallery)
-
-    gr.Markdown(
-        """___
-      <p style='text-align: center
-      '>
-      Created by https://huggingface.co/hakurei
-      <br/>
-      </p>"""
-    )
-
-demo.launch(debug=True,
-            server_port=7860)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+
+def __def_helper():
+    StableDiffusionPipeline.__call__()
+
+
+pipe = StableDiffusionPipeline.from_pretrained(model_id,
+                                               resume_download=True,  # 模型文件断点续传
+                                               torch_dtype=torch.float16,
+                                               revision='fp16')
+pipe = pipe.to(device)
+
+
+def infer(prompt, width, height, nums, steps, guidance_scale, seed):
+    print(prompt, width, height, nums, steps, guidance_scale, seed)
+
+    if prompt is not None and prompt != "":
+
+        if seed is None or seed == '' or seed == -1:
+            seed = int(random.randrange(4294967294))
+
+        with autocast("cuda"):
+            generator = torch.Generator("cuda").manual_seed(seed)
+
+            images = pipe([prompt] * nums,
+                          height=height,
+                          width=width,
+                          num_inference_steps=steps,
+                          generator=generator,
+                          guidance_scale=guidance_scale
+                          )["sample"]
+            return images
+
+
+description = """
+prompt 素材：[https://lexica.art](https://lexica.art) \n
+seed：为空会使用随机seed
+
+"""
+
+
+# with block as demo:
+def run():
+    _app = gr.Interface(
+        fn=infer,
+        title="Waifu Diffusion",
+        description=description,
+        inputs=[
+            gr.Textbox(label="输入 prompt"),
+            gr.Slider(512, 1024, 512, step=64, label="width"),
+            gr.Slider(512, 1024, 512, step=64, label="height"),
+            gr.Slider(1, 4, 1, step=1, label="Number of Images"),
+            gr.Slider(10, 150, step=1, value=50,
+                      label="num_inference_steps"),
+            gr.Slider(0, 20, 7.5, step=0.5,
+                      label="guidance_scale:\n" +
+                            "较高的引导比例鼓励生成与文本“提示”密切相关的图像，通常以降低图像质量为代价"),
+            gr.Textbox(label="随机 send",
+                       placeholder="Random Seed",
+                       lines=1),
+        ],
+        outputs=[
+            gr.Gallery(label="Generated images")
+        ])
+
+    return _app
+
+
+app = run()
+app.launch(debug=True)

app2.py

@@ -0,0 +1,81 @@
+# pip install transformers gradio scipy ftfy "ipywidgets>=7,<8" datasets diffusers
+
+import gradio as gr
+import torch
+from torch import autocast
+from diffusers import StableDiffusionPipeline
+
+model_id = "hakurei/waifu-diffusion"
+device = "cpu"
+
+# pipe = StableDiffusionPipeline.from_pretrained(model_id,
+#                                                resume_download=True,  # 模型文件断点续传
+#                                                torch_dtype=torch.float16,
+#                                                revision='fp16')
+# pipe = pipe.to(device)
+
+block = gr.Blocks(css=".container { max-width: 800px; margin: auto; }")
+
+num_samples = 2
+
+
+def infer(prompt):
+    print(prompt)
+    return prompt
+    # with autocast("cuda"):
+    #     images = pipe([prompt] * num_samples,
+    #                   hight=111,
+    #                   width=100,
+    #
+    #                   guidance_scale=7.5)["sample"]
+
+    # return images
+
+
+with block as demo:
+    gr.Markdown("<h1><center>Waifu Diffusion</center></h1>")
+    gr.Markdown(
+        "waifu-diffusion is a latent text-to-image diffusion model that has been conditioned on high-quality anime images through fine-tuning."
+    )
+    with gr.Group():
+        with gr.Box():
+            with gr.Column().style(mobile_collapse=False, equal_height=True):
+                text = gr.Textbox(
+                    label="Enter your prompt",
+                    show_label=False,
+                    max_lines=1
+                ).style(
+                    border=(True, False, True, True),
+                    rounded=(True, False, False, True),
+                    container=False,
+                )
+                slider = gr.Slider(0, 1000, 10)
+
+                btn = gr.Button("Run").style(
+                    margin=False,
+                    rounded=(False, True, True, False),
+                )
+
+        gallery = gr.Gallery(
+            label="Generated images",
+            show_label=False
+        ).style(
+            grid=[2],
+            height="auto"
+        )
+        gr.Interface
+        text.submit(infer,
+                    inputs=[text,slider] ,
+                    outputs=gallery)
+        btn.click(infer, inputs=[text], outputs=gallery)
+
+    gr.Markdown(
+        """___
+      <p style='text-align: center
+      '>
+      Created by https://huggingface.co/hakurei
+      <br/>
+      </p>"""
+    )
+
+demo.launch(debug=True)