jpjp9292 commited on
Commit
ce40f75
ยท
verified ยท
1 Parent(s): 41204f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +39 -36
app.py CHANGED
@@ -19,20 +19,19 @@ MAX_SEED = np.iinfo(np.int32).max
19
  MAX_IMAGE_SIZE = 1024
20
 
21
  def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
22
-
23
  if randomize_seed:
24
  seed = random.randint(0, MAX_SEED)
25
 
26
  generator = torch.Generator().manual_seed(seed)
27
 
28
  image = pipe(
29
- prompt = prompt,
30
- negative_prompt = negative_prompt,
31
- guidance_scale = guidance_scale,
32
- num_inference_steps = num_inference_steps,
33
- width = width,
34
- height = height,
35
- generator = generator
36
  ).images[0]
37
 
38
  return image
@@ -41,9 +40,12 @@ examples = [
41
  "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
42
  "An astronaut riding a green horse",
43
  "A delicious ceviche cheesecake slice",
 
 
 
44
  ]
45
 
46
- css="""
47
  #col-container {
48
  margin: 0 auto;
49
  max-width: 520px;
@@ -60,87 +62,88 @@ with gr.Blocks(css=css) as demo:
60
  with gr.Column(elem_id="col-container"):
61
  gr.Markdown(f"""
62
  # Text-to-Image Gradio Template
 
63
  Currently running on {power_device}.
 
64
  """)
65
 
66
  with gr.Row():
67
-
68
- prompt = gr.Text(
69
- label="Prompt",
70
  show_label=False,
71
  max_lines=1,
72
- placeholder="Enter your prompt",
73
  container=False,
74
  )
75
 
76
- run_button = gr.Button("Run", scale=0)
77
 
78
- result = gr.Image(label="Result", show_label=False)
79
 
80
- with gr.Accordion("Advanced Settings", open=False):
81
 
82
- negative_prompt = gr.Text(
83
- label="Negative prompt",
84
  max_lines=1,
85
- placeholder="Enter a negative prompt",
86
  visible=False,
87
  )
88
 
89
  seed = gr.Slider(
90
- label="Seed",
91
  minimum=0,
92
  maximum=MAX_SEED,
93
  step=1,
94
  value=0,
95
  )
96
 
97
- randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
98
 
99
  with gr.Row():
100
 
101
  width = gr.Slider(
102
- label="Width",
103
  minimum=256,
104
  maximum=MAX_IMAGE_SIZE,
105
  step=32,
106
- value=512,
107
  )
108
 
109
  height = gr.Slider(
110
- label="Height",
111
  minimum=256,
112
  maximum=MAX_IMAGE_SIZE,
113
  step=32,
114
- value=512,
115
  )
116
 
117
  with gr.Row():
118
 
119
  guidance_scale = gr.Slider(
120
- label="Guidance scale",
121
  minimum=0.0,
122
  maximum=10.0,
123
  step=0.1,
124
- value=0.0,
125
  )
126
 
127
  num_inference_steps = gr.Slider(
128
- label="Number of inference steps",
129
  minimum=1,
130
- maximum=12,
131
  step=1,
132
- value=2,
133
  )
134
 
135
  gr.Examples(
136
- examples = examples,
137
- inputs = [prompt]
138
  )
139
 
140
  run_button.click(
141
- fn = infer,
142
- inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
143
- outputs = [result]
144
  )
145
 
146
- demo.queue().launch()
 
19
  MAX_IMAGE_SIZE = 1024
20
 
21
  def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
 
22
  if randomize_seed:
23
  seed = random.randint(0, MAX_SEED)
24
 
25
  generator = torch.Generator().manual_seed(seed)
26
 
27
  image = pipe(
28
+ prompt=prompt,
29
+ negative_prompt=negative_prompt,
30
+ guidance_scale=guidance_scale,
31
+ num_inference_steps=num_inference_steps,
32
+ width=width,
33
+ height=height,
34
+ generator=generator
35
  ).images[0]
36
 
37
  return image
 
40
  "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
41
  "An astronaut riding a green horse",
42
  "A delicious ceviche cheesecake slice",
43
+ "์ •๊ธ€์— ์žˆ๋Š” ์šฐ์ฃผ ๋น„ํ–‰์‚ฌ, ์ฐจ๊ฐ€์šด ์ƒ‰ ํŒ”๋ ˆํŠธ, muted ์ƒ‰์ƒ, ์ƒ์„ธํ•œ, 8k",
44
+ "๋…น์ƒ‰ ๋ง์„ ํƒ€๊ณ  ์žˆ๋Š” ์šฐ์ฃผ ๋น„ํ–‰์‚ฌ",
45
+ "๋ง›์žˆ๋Š” ์„ธ๋น„์ฒด ์น˜์ฆˆ์ผ€์ดํฌ ์กฐ๊ฐ",
46
  ]
47
 
48
+ css = """
49
  #col-container {
50
  margin: 0 auto;
51
  max-width: 520px;
 
62
  with gr.Column(elem_id="col-container"):
63
  gr.Markdown(f"""
64
  # Text-to-Image Gradio Template
65
+ # ํ…์ŠคํŠธ-์ด๋ฏธ์ง€ ์ƒ์„ฑ๊ธฐ Gradio ํ…œํ”Œ๋ฆฟ
66
  Currently running on {power_device}.
67
+ ํ˜„์žฌ {power_device}์—์„œ ์‹คํ–‰ ์ค‘์ž…๋‹ˆ๋‹ค.
68
  """)
69
 
70
  with gr.Row():
71
+ prompt = gr.Textbox(
72
+ label="Prompt / ํ”„๋กฌํ”„ํŠธ",
 
73
  show_label=False,
74
  max_lines=1,
75
+ placeholder="Enter your prompt / ํ”„๋กฌํ”„ํŠธ๋ฅผ ์ž…๋ ฅํ•˜์„ธ์š”",
76
  container=False,
77
  )
78
 
79
+ run_button = gr.Button("Run / ์‹คํ–‰", scale=0)
80
 
81
+ result = gr.Image(label="Result / ๊ฒฐ๊ณผ", show_label=False)
82
 
83
+ with gr.Accordion("Advanced Settings / ๊ณ ๊ธ‰ ์„ค์ •", open=False):
84
 
85
+ negative_prompt = gr.Textbox(
86
+ label="Negative prompt / ๋„ค๊ฑฐํ‹ฐ๋ธŒ ํ”„๋กฌํ”„ํŠธ",
87
  max_lines=1,
88
+ placeholder="Enter a negative prompt / ๋„ค๊ฑฐํ‹ฐ๋ธŒ ํ”„๋กฌํ”„ํŠธ๋ฅผ ์ž…๋ ฅํ•˜์„ธ์š”",
89
  visible=False,
90
  )
91
 
92
  seed = gr.Slider(
93
+ label="Seed / ์‹œ๋“œ",
94
  minimum=0,
95
  maximum=MAX_SEED,
96
  step=1,
97
  value=0,
98
  )
99
 
100
+ randomize_seed = gr.Checkbox(label="Randomize seed / ์‹œ๋“œ ๋žœ๋คํ™”", value=True)
101
 
102
  with gr.Row():
103
 
104
  width = gr.Slider(
105
+ label="Width / ๋„ˆ๋น„",
106
  minimum=256,
107
  maximum=MAX_IMAGE_SIZE,
108
  step=32,
109
+ value=1024,
110
  )
111
 
112
  height = gr.Slider(
113
+ label="Height / ๋†’์ด",
114
  minimum=256,
115
  maximum=MAX_IMAGE_SIZE,
116
  step=32,
117
+ value=1024,
118
  )
119
 
120
  with gr.Row():
121
 
122
  guidance_scale = gr.Slider(
123
+ label="Guidance scale / ๊ฐ€์ด๋˜์Šค ์Šค์ผ€์ผ",
124
  minimum=0.0,
125
  maximum=10.0,
126
  step=0.1,
127
+ value=7.5,
128
  )
129
 
130
  num_inference_steps = gr.Slider(
131
+ label="Number of inference steps / ์ถ”๋ก  ๋‹จ๊ณ„ ์ˆ˜",
132
  minimum=1,
133
+ maximum=50,
134
  step=1,
135
+ value=25,
136
  )
137
 
138
  gr.Examples(
139
+ examples=examples,
140
+ inputs=[prompt]
141
  )
142
 
143
  run_button.click(
144
+ fn=infer,
145
+ inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
146
+ outputs=[result]
147
  )
148
 
149
+ demo.queue().launch()