Kevin676 commited on
Commit
04b3352
·
1 Parent(s): 5bad1f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -3
app.py CHANGED
@@ -4,6 +4,16 @@ import torch
4
  from langs import LANGS
5
  import os
6
 
 
 
 
 
 
 
 
 
 
 
7
  pipe1 = pipeline('text-generation', model='RamAnanth1/distilgpt2-sd-prompts')
8
 
9
  TASK = "translation"
@@ -54,6 +64,14 @@ def infer(input):
54
  result = get_new_prompt(img, 'fast')
55
  return result[0]
56
 
 
 
 
 
 
 
 
 
57
  with gr.Blocks() as demo:
58
  gr.Markdown(
59
  """ # <center>🥳💬💕 - Stable Diffusion,随时随地,想画就画!</center>
@@ -73,14 +91,27 @@ with gr.Blocks() as demo:
73
  with gr.Row().style(equal_height=True):
74
  inp2 = out1
75
  btn1 = gr.Button("让您的提示词更详细一些吧")
76
- out2 = gr.Textbox(label = "翻译后的英文内容", lines=1)
77
 
78
  btn1.click(infer, [inp2], [out2])
79
-
 
 
 
 
 
 
 
 
 
 
 
 
 
80
  gr.Markdown(
81
  """ ### <center>注意❗:请不要输入或生成会对个人以及组织造成侵害的内容,此程序仅供科研、学习及娱乐使用。用户输入或生成的内容与程序开发者无关,请自觉合法合规使用,违反者一切后果自负。</center>
82
 
83
- ### <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>
84
 
85
  """
86
  )
 
4
  from langs import LANGS
5
  import os
6
 
7
+ import numpy as np
8
+ import modin.pandas as pd
9
+ from PIL import Image
10
+ from diffusers import DiffusionPipeline #EulerDiscreteScheduler
11
+
12
+ device1 = "cuda" if torch.cuda.is_available() else "cpu"
13
+
14
+ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", safety_checker=None)
15
+ pipe = pipe.to(device1)
16
+
17
  pipe1 = pipeline('text-generation', model='RamAnanth1/distilgpt2-sd-prompts')
18
 
19
  TASK = "translation"
 
64
  result = get_new_prompt(img, 'fast')
65
  return result[0]
66
 
67
+ #stable diffusion
68
+
69
+ def genie (prompt, negative_prompt, scale, steps, seed):
70
+ generator = torch.Generator(device=device1).manual_seed(seed)
71
+ 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]
72
+ return images
73
+
74
+
75
  with gr.Blocks() as demo:
76
  gr.Markdown(
77
  """ # <center>🥳💬💕 - Stable Diffusion,随时随地,想画就画!</center>
 
91
  with gr.Row().style(equal_height=True):
92
  inp2 = out1
93
  btn1 = gr.Button("让您的提示词更详细一些吧")
94
+ out2 = gr.Textbox(label = "更详细的提示词(可在此编辑)", lines=1)
95
 
96
  btn1.click(infer, [inp2], [out2])
97
+
98
+ with gr.Row().style(equal_height=True):
99
+ inp3 = out2
100
+ inp4 = gr.Textbox(label="删去一些您不想要的特点(反向关键词;选填)", placeholder = "low quality", lines=1)
101
+ inp5 = gr.Slider(1, 25, 10)
102
+ inp6 = gr.Slider(30, maximum=75, value=50, step=1)
103
+ inp7 = gr.Slider(minimum=1, step=1, maximum=999999999999999999, randomize=True)
104
+
105
+ with gr.Row().style(equal_height=True):
106
+ btn2 = gr.Button("开始生成图片吧")
107
+ out3 = gr.Image(label="为您生成的专属图片")
108
+
109
+ btn2.click(genie, [inp3, inp4, inp5, inp6, inp7], [out3])
110
+
111
  gr.Markdown(
112
  """ ### <center>注意❗:请不要输入或生成会对个人以及组织造成侵害的内容,此程序仅供科研、学习及娱乐使用。用户输入或生成的内容与程序开发者无关,请自觉合法合规使用,违反者一切后果自负。</center>
113
 
114
+ ### <center>Model by [Stable Diffusion](https://github.com/Stability-AI/stablediffusion). Thanks to [Manjushri](https://huggingface.co/Manjushri).</center>
115
 
116
  """
117
  )