Update app.py

app.py

@@ -212,9 +212,14 @@ class ModelManager:
             "anime_aesthetic": {"name": "Anime Score", "process": self._process_anime_aesthetic, "model": self.anime_aesthetic_model},
         }
         self.processing_queue: asyncio.Queue = asyncio.Queue()
-        self.worker_task = asyncio.create_task(self._worker())
+        self.worker_task = None # Initialize worker_task to None
         self.temp_dir = tempfile.mkdtemp()
 
+    async def start_worker(self):
+        """Start the background worker task."""
+        if self.worker_task is None:
+            self.worker_task = asyncio.create_task(self._worker())
+
     async def _worker(self):
         """Background worker to process image evaluation requests from the queue."""
         while True:
@@ -494,13 +499,15 @@ class ModelManager:
         """Clean up temporary directories and shutdown worker."""
         if os.path.exists(self.temp_dir):
             shutil.rmtree(self.temp_dir)
-        asyncio.run(self.shutdown()) # Shutdown worker gracefully
+        if self.worker_task is not None: # Check if worker_task was started
+            asyncio.run(self.shutdown()) # Shutdown worker gracefully
 
     async def shutdown(self):
         """Send shutdown signal to worker and wait for it to finish."""
-        await self.processing_queue.put(None) # Send shutdown signal
-        await self.worker_task # Wait for worker task to complete
-        await self.processing_queue.join() # Wait for queue to be empty
+        if self.worker_task is not None: # Check if worker_task was started
+            await self.processing_queue.put(None) # Send shutdown signal
+            await self.worker_task # Wait for worker task to complete
+            await self.processing_queue.join() # Wait for queue to be empty
 
 
 #####################################
@@ -683,6 +690,8 @@ def create_interface():
             outputs=[download_file_output] # Output is now the gr.File component
         )
         demo.load(lambda: update_table_sort("Final Score", model_options, []), inputs=None, outputs=[output_html, global_results_state]) # Initial sort and table render, pass empty initial results
+        demo.load(model_manager.start_worker) # Start the worker task on demo load
+
         gr.Markdown("""
         ### Notes
         - Select models to use for evaluation using the checkboxes.