Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
3790166
1
Parent(s):
82d3087
sdf
Browse files
app.py
CHANGED
@@ -25,6 +25,8 @@ def on_upload() -> None:
|
|
25 |
del st.session_state['seg']
|
26 |
if 'unique_colors' in st.session_state:
|
27 |
del st.session_state['unique_colors']
|
|
|
|
|
28 |
|
29 |
|
30 |
def check_reset_state() -> bool:
|
|
|
25 |
del st.session_state['seg']
|
26 |
if 'unique_colors' in st.session_state:
|
27 |
del st.session_state['unique_colors']
|
28 |
+
if 'output_image' in st.session_state:
|
29 |
+
del st.session_state['output_image']
|
30 |
|
31 |
|
32 |
def check_reset_state() -> bool:
|
models.py
CHANGED
@@ -249,7 +249,7 @@ def make_image_controlnet(image: np.ndarray,
|
|
249 |
|
250 |
with catchtime("Controlnet generation total"):
|
251 |
for _, setting in enumerate(prompt_settings):
|
252 |
-
st.success(f"{pipe.queue_size} images in the queue, can take up to {pipe.queue_size * 20} seconds")
|
253 |
with catchtime("Controlnet generation"):
|
254 |
generated_image = pipe(
|
255 |
**setting,
|
|
|
249 |
|
250 |
with catchtime("Controlnet generation total"):
|
251 |
for _, setting in enumerate(prompt_settings):
|
252 |
+
st.success(f"{pipe.queue_size} images in the queue, can take up to {(pipe.queue_size)+1 * 20} seconds")
|
253 |
with catchtime("Controlnet generation"):
|
254 |
generated_image = pipe(
|
255 |
**setting,
|