queue

models.py

@@ -49,6 +49,10 @@ class ControlNetPipeline:
         self.waiting_queue = []
         self.count = 0
     
+    @property
+    def queue_size(self):
+        return len(self.waiting_queue)
+    
     def __call__(self, **kwargs):
         self.count += 1
         number = self.count
@@ -245,6 +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")
             with catchtime("Controlnet generation"):
                 generated_image = pipe(
                     **setting,