Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
f64d68f
1
Parent(s):
8cef0af
up
Browse files
app.py
CHANGED
@@ -127,7 +127,7 @@ def txt_to_img(
|
|
127 |
pipe.to("cpu")
|
128 |
torch.cuda.empty_cache()
|
129 |
|
130 |
-
return low_res_image, up_res_image
|
131 |
|
132 |
|
133 |
def replace_nsfw_images(results):
|
@@ -176,13 +176,8 @@ with gr.Blocks(css="style.css") as demo:
|
|
176 |
rounded=(False, True, True, False)
|
177 |
)
|
178 |
|
179 |
-
low_res_image = gr.
|
180 |
-
|
181 |
-
).style(grid=[2], height=512)
|
182 |
-
|
183 |
-
up_res_image = gr.Gallery(
|
184 |
-
label="1024-pix image", show_label=False, elem_id="gallery"
|
185 |
-
).style(grid=[2], height=1024)
|
186 |
|
187 |
state_info = gr.Textbox(label="State", show_label=False, max_lines=2).style(
|
188 |
container=False
|
|
|
127 |
pipe.to("cpu")
|
128 |
torch.cuda.empty_cache()
|
129 |
|
130 |
+
return low_res_image[0], up_res_image[0]
|
131 |
|
132 |
|
133 |
def replace_nsfw_images(results):
|
|
|
176 |
rounded=(False, True, True, False)
|
177 |
)
|
178 |
|
179 |
+
low_res_image = gr.Image(label="512px Image")
|
180 |
+
up_res_image = gr.Image(label="1024px Image")
|
|
|
|
|
|
|
|
|
|
|
181 |
|
182 |
state_info = gr.Textbox(label="State", show_label=False, max_lines=2).style(
|
183 |
container=False
|