sdf

app.py

@@ -25,6 +25,8 @@ def on_upload() -> None:
             del st.session_state['seg']
         if 'unique_colors' in st.session_state:
             del st.session_state['unique_colors']
+        if 'output_image' in st.session_state:
+            del st.session_state['output_image']
 
 
 def check_reset_state() -> bool:

models.py

@@ -249,7 +249,7 @@ def make_image_controlnet(image: np.ndarray,
 
     with catchtime("Controlnet generation total"):
         for _, setting in enumerate(prompt_settings):
-            st.success(f"{pipe.queue_size} images in the queue, can take up to {pipe.queue_size * 20} seconds")
+            st.success(f"{pipe.queue_size} images in the queue, can take up to {(pipe.queue_size)+1 * 20} seconds")
             with catchtime("Controlnet generation"):
                 generated_image = pipe(
                     **setting,