add KoolCogVideoX

README.md

@@ -1,5 +1,5 @@
 ---
-title: VideoSys-CogVideoX
+title: KoolCogVideoX
 emoji: 🎥
 colorFrom: yellow
 colorTo: green
@@ -10,8 +10,8 @@ suggested_storage: large
 app_port: 7860
 app_file: app.py
 models:
-  - THUDM/CogVideoX-2b
-  - THUDM/CogVideoX-5b
+  - bertjiazheng/KoolCogVideoX-2b
+  - bertjiazheng/KoolCogVideoX-2b
 tags:
   - cogvideox
   - video-generation

app.py

@@ -10,6 +10,7 @@ import spaces
 
 from videosys import CogVideoXConfig, CogVideoXPABConfig, VideoSysEngine
 
+PROMPT = "A modern living room with a minimalist design, featuring a large window, a white ceiling, and a wooden floor. The room is furnished with a white sofa, a gray ottoman, a wooden table, and a hanging light. The space is well-lit and has a clean, contemporary aesthetic."
 
 def load_model(model_name, enable_video_sys=False, pab_threshold=[100, 850], pab_range=2):
     pab_config = CogVideoXPABConfig(spatial_threshold=pab_threshold, spatial_range=pab_range)
@@ -131,29 +132,27 @@ with gr.Blocks(css=css) as demo:
     gr.HTML(
         """
     <div style="text-align: center; font-size: 32px; font-weight: bold; margin-bottom: 20px;">
-        VideoSys for CogVideoX🤗
+        KoolCogVideoX Huggingface Space🤗
+    </div>
+    <div style="text-align: center; font-size: 15px; margin-bottom: 20px;">
+        KoolCogVideoX is fine-tuned on <a href="https://huggingface.co/collections/THUDM/cogvideo-66c08e62f1685a3ade464cce">CogVideoX</a> specifically for interior design scenarios.<br>
+        The demo is powered by <a href="https://github.com/NUS-HPC-AI-Lab/VideoSys">https://github.com/NUS-HPC-AI-Lab/VideoSys</a>.
     </div>
     <div style="text-align: center; font-size: 15px;">
-        🌐 Github: <a href="https://github.com/NUS-HPC-AI-Lab/VideoSys">https://github.com/NUS-HPC-AI-Lab/VideoSys</a><br>
-
-        ⚠️ This demo is for academic research and experiential use only.
-        Users should strictly adhere to local laws and ethics.<br>
-
-        💡 This demo only demonstrates single-device inference. To experience the full power of VideoSys, please deploy it with multiple devices.<br><br>
-        </div>
+        ⚠️ This demo is for academic research and experiential use only. Users should strictly adhere to local laws and ethics.
     </div>
     """
     )
 
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="Prompt (Less than 200 Words)", value="Sunset over the sea.", lines=2)
+            prompt = gr.Textbox(label="Prompt (Less than 200 Words)", value=PROMPT, lines=2)
 
             with gr.Column():
                 gr.Markdown("**Generation Parameters**<br>")
                 with gr.Row():
                     model_name = gr.Radio(
-                        ["THUDM/CogVideoX-2b", "THUDM/CogVideoX-5b"], label="Model Type", value="THUDM/CogVideoX-2b"
+                        ["bertjiazheng/KoolCogVideoX-2b", "bertjiazheng/KoolCogVideoX-5b"], label="Model Type", value="bertjiazheng/KoolCogVideoX-5b"
                     )
                 with gr.Row():
                     num_inference_steps = gr.Slider(label="Inference Steps", maximum=50, value=50)
@@ -191,6 +190,43 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 video_output_vs = gr.Video(label="CogVideoX with VideoSys", width=720, height=480)
 
+    gr.Markdown("""
+    <table border="0" style="width: 100%; text-align: left; margin-top: 20px;">
+        <div style="text-align: center; font-size: 32px; font-weight: bold; margin-bottom: 20px;">
+            🎥 Video Gallery
+        </div>
+        <tr>
+            <td style="width: 25%; vertical-align: top; font-size: 0.9em;">
+                <p>A modern living room with a minimalist design, featuring a white sofa, a marble coffee table, a geometric painting, and a chandelier hanging from the ceiling. The room is well-lit with natural light, and the color scheme is neutral with accents of gold and black. The furniture is arranged in a way that creates a comfortable and inviting space.
+            </p>
+            </td>
+            <td style="width: 25%; vertical-align: top;">
+                <video src="https://manycore-research-azure.kujiale.com/manycore-research/KoolCogVideoX-5b/L3D386S81B20ENDPDZEFIAUWJLWGLUFX7LB6KY8.mp4" width="100%" controls autoplay loop></video>
+            </td>
+            <td style="width: 25%; vertical-align: top; font-size: 0.9em;">
+                <p>A modern living room with a minimalist design, featuring a large window, a white ceiling, and a wooden floor. The room is furnished with a white sofa, a gray ottoman, a wooden table, and a hanging light. The space is well-lit and has a clean, contemporary aesthetic.</p>
+            </td>
+            <td style="width: 25%; vertical-align: top;">
+                <video src="https://manycore-research-azure.kujiale.com/manycore-research/KoolCogVideoX-5b/L3D489S267B0ENDPN7LV6QUWLYIALUFX4RSVTA8.mp4" width="100%" controls autoplay loop></video>
+            </td>
+        </tr>
+        <tr>
+            <td style="width: 25%; vertical-align: top; font-size: 0.9em;">
+                <p>A modern bedroom with a minimalist design, featuring a large bed with a gray comforter and a blue blanket, a white dresser with a mirror, and a white closet. The room is decorated with framed artwork and a black and white poster on the wall. The floor is made of light wood, and the room has a clean and contemporary feel.</p>
+            </td>
+            <td style="width: 25%; vertical-align: top;">
+                <video src="https://manycore-research-azure.kujiale.com/manycore-research/KoolCogVideoX-5b/L3D943S369B0ENDPNXSMXYUWLGWQLUFX6ZZZWA8.mp4" width="100%" controls autoplay loop></video>
+            </td>
+            <td style="width: 25%; vertical-align: top; font-size: 0.9em;">
+                <p>A modern kitchen with a sleek design, featuring a marble countertop, stainless steel appliances, and a variety of bottles and glasses. The kitchen is well-lit with recessed lighting and has a contemporary aesthetic.</p>
+            </td>
+            <td style="width: 25%; vertical-align: top;">
+                <video src="https://manycore-research-azure.kujiale.com/manycore-research/KoolCogVideoX-5b/L3D670S366B0ENDPOAJ4LQUWJN2ILUFX5GLPRQ8.mp4" width="100%" controls autoplay loop></video>
+            </td>
+        </tr>
+    </table>
+        """)
+
     generate_button_vs.click(
         generate_vs,
         inputs=[