Skip to content

Commit a15cb14

Browse files
committed
[WEB] Display the 512x512 image size
Signed-Off-by: Gaurav Shukla <[email protected]>
1 parent 898bc9e commit a15cb14

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

web/index.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def resource_path(relative_path):
5050
footer {display: none !important;}
5151
"""
5252

53-
with gr.Blocks(css=demo_css) as shark_web:
53+
with gr.Blocks(title="Stable Diffusion", css=demo_css) as shark_web:
5454

5555
with gr.Row(elem_id="ui_title"):
5656
nod_logo = Image.open(nodlogo_loc)
@@ -139,16 +139,16 @@ def resource_path(relative_path):
139139
_js="(min,max) => Math.floor(Math.random() * (max - min)) + min",
140140
)
141141
stable_diffusion = gr.Button("Generate Image")
142-
with gr.Accordion("Performace Details:"):
142+
with gr.Column(scale=1, min_width=600):
143+
with gr.Group():
144+
generated_img = gr.Image(
145+
type="pil", interactive=False
146+
).style(height=512)
143147
std_output = gr.Textbox(
144148
value="Nothing to show.",
145149
lines=4,
146150
show_label=False,
147151
)
148-
with gr.Column(scale=1, min_width=600):
149-
generated_img = gr.Image(type="pil", interactive=False).style(
150-
height=768
151-
)
152152

153153
prompt.submit(
154154
stable_diff_inf,
@@ -178,6 +178,7 @@ def resource_path(relative_path):
178178
shark_web.queue()
179179
shark_web.launch(
180180
share=False,
181+
inbrowser=True,
181182
server_name="0.0.0.0",
182183
server_port=8080,
183184
)

0 commit comments

Comments
 (0)