Adding final progress

app.py

@@ -263,12 +263,12 @@ def generate_topics(dataset, config, split, column, nested_column):
         offset += CHUNK_SIZE
 
     logging.info("Finished processing all data")
-    cuda.empty_cache()  # Clear cache at the end of each chunk
-    return (
+    yield (
         topics_info,
         topic_plot,
-        gr.Label({f"⚙️ Generating topics {dataset}": 1.0}, visible=True),
+        gr.Label({f"✅ Generating topics {dataset}": 1.0}, visible=True),
     )
+    cuda.empty_cache()
 
 
 with gr.Blocks() as demo: