Update app.py

app.py

@@ -4,6 +4,16 @@ import torch
 from langs import LANGS
 import os
 
+import numpy as np
+import modin.pandas as pd
+from PIL import Image
+from diffusers import DiffusionPipeline #EulerDiscreteScheduler
+
+device1 = "cuda" if torch.cuda.is_available() else "cpu"
+
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", safety_checker=None)
+pipe = pipe.to(device1)
+
 pipe1 = pipeline('text-generation', model='RamAnanth1/distilgpt2-sd-prompts')
 
 TASK = "translation"
@@ -54,6 +64,14 @@ def infer(input):
     result = get_new_prompt(img, 'fast')    
     return result[0]
 
+#stable diffusion
+
+def genie (prompt, negative_prompt, scale, steps, seed):
+     generator = torch.Generator(device=device1).manual_seed(seed)
+     images = pipe(prompt, negative_prompt=negative_prompt, width=768, height=768, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator).images[0]
+     return images
+
+
 with gr.Blocks() as demo:
     gr.Markdown(
             """ # <center>🥳💬💕 - Stable Diffusion，随时随地，想画就画！</center>
@@ -73,14 +91,27 @@ with gr.Blocks() as demo:
     with gr.Row().style(equal_height=True):
         inp2 = out1
         btn1 = gr.Button("让您的提示词更详细一些吧")
-        out2 = gr.Textbox(label = "翻译后的英文内容", lines=1)
+        out2 = gr.Textbox(label = "更详细的提示词(可在此编辑)", lines=1)
     
     btn1.click(infer, [inp2], [out2])
-   
+
+    with gr.Row().style(equal_height=True):
+        inp3 = out2
+        inp4 = gr.Textbox(label="删去一些您不想要的特点(反向关键词；选填)", placeholder = "low quality", lines=1)
+        inp5 = gr.Slider(1, 25, 10)
+        inp6 = gr.Slider(30, maximum=75, value=50, step=1)
+        inp7 = gr.Slider(minimum=1, step=1, maximum=999999999999999999, randomize=True)
+
+    with gr.Row().style(equal_height=True):
+        btn2 = gr.Button("开始生成图片吧")
+        out3 = gr.Image(label="为您生成的专属图片")
+
+    btn2.click(genie, [inp3, inp4, inp5, inp6, inp7], [out3])
+    
     gr.Markdown(
             """ ### <center>注意❗：请不要输入或生成会对个人以及组织造成侵害的内容，此程序仅供科研、学习及娱乐使用。用户输入或生成的内容与程序开发者无关，请自觉合法合规使用，违反者一切后果自负。</center>
             
-            ### <center>Model by [ChatGLM-6B](https://huggingface.co/THUDM/chatglm-6b). Thanks to [THUDM](https://github.com/THUDM) and [CjangCjengh](https://github.com/CjangCjengh). Please follow me on [Bilibili](https://space.bilibili.com/501495851?spm_id_from=333.1007.0.0).</center>
+            ### <center>Model by [Stable Diffusion](https://github.com/Stability-AI/stablediffusion). Thanks to [Manjushri](https://huggingface.co/Manjushri).</center>
             
         """
     )