fixed examples

demo.py

@@ -667,6 +667,43 @@ def load_view_mask(view):
         return None
 
 
+custom_js = """
+<script>
+console.log("Hello, world!");
+(function() {
+    // Poll every 100ms for the existence of the header row
+    const intervalId = setInterval(() => {
+        console.log("Polling for header row");
+        const headerRow = document.querySelector("tr.tr-head");
+        if (headerRow) {
+            const headers = headerRow.querySelectorAll("th");
+            headers.forEach(cell => {
+                const text = cell.innerText.trim();
+                if (text === "Binary Mask") {
+                    cell.innerText = "Mask";
+                } else if (text === "View Class") {
+                    cell.innerText = "View";
+                } else if (text === "Number of Sampling Steps") {
+                    cell.innerText = "Img Samp. Steps";
+                } else if (text === "Ejection Fraction (%)") {
+                    cell.innerText = "EF %";
+                } else if (text === "Number of Sampling Steps.") {
+                    cell.innerText = "Video Samp. Steps";
+                } else if (text === "Classifier-Free Guidance Scale") {
+                    cell.innerText = "CFG";
+                } else if (text === "Filtered Latent Image") {
+                    cell.innerText = "Filtered Image";
+                }
+            });
+            clearInterval(intervalId);
+            console.log("Headers updated.");
+        }
+    }, 500);
+})();
+</script>
+"""
+
+
 def create_demo():
     # Define all components first
     mask_input = gr.ImageEditor(
@@ -719,7 +756,7 @@ def create_demo():
         maximum=200,
         value=100,
         step=1,
-        label="Number of Sampling Steps",
+        label="Number of Sampling Steps.",
         render=False,
     )
 
@@ -762,16 +799,20 @@ def create_demo():
         label="Latent Video",
         format="mp4",
         render=False,
+        autoplay=True,
+        loop=True,
     )
 
     decoded_animation_display = gr.Video(
         label="Decoded Video",
         format="mp4",
         render=False,
+        autoplay=True,
+        loop=True,
     )
 
     # Define the theme and layout
-    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    with gr.Blocks(theme=gr.themes.Soft(), head=custom_js) as demo:
         gr.Markdown("# EchoFlow Demo")
         gr.Markdown("## Dataset Generation Pipeline")