update script

app.py

@@ -1,7 +1,75 @@
+import os
+
+if os.environ.get("SPACES_ZERO_GPU") is not None:
+    import spaces
+else:
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+
+            return wrapper
+
+import torch
+from diffusers import MochiPipeline
+from diffusers.utils import export_to_video
 import gradio as gr
+import config as cfg
+
+# Load the pre-trained model
+pipe = MochiPipeline.from_pretrained(cfg.MODEL_PRE_TRAINED_ID, variant="bf16", torch_dtype=torch.bfloat16)
+
+# Enable memory-saving optimizations
+pipe.enable_model_cpu_offload()
+pipe.enable_vae_tiling()
+
+
+@spaces.GPU(duration=240)
+def generate_video(prompt, num_frames=84, fps=30, high_quality=False):
+    if high_quality:
+        print("High quality option selected. Requires 42GB VRAM.")
+        # Check if running on ZeroGPU
+        if os.environ.get("SPACES_ZERO_GPU") is not None:
+            raise RuntimeError("High quality option may fail on ZeroGPU environments.")
+        with torch.autocast("cuda", torch.bfloat16, cache_enabled=False):
+            frames = pipe(prompt, num_frames=num_frames).frames[0]
+    else:
+        print("Standard quality option selected.")
+        frames = pipe(prompt, num_frames=num_frames).frames[0]
+
+        # Export frames as video
+    video_path = "mochi.mp4"
+    export_to_video(frames, video_path, fps=fps)
+    return video_path
+
+
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=generate_video,
+    inputs=[
+        gr.Textbox(lines=2, placeholder="Enter your text prompt here... 💡"),
+        gr.Slider(minimum=1, maximum=240, value=84, label="Number of frames 🎞️"),
+        gr.Slider(minimum=1, maximum=60, value=30, label="FPS (Frames per second) ⏱️"),
+        gr.Checkbox(label="High Quality Output (requires 42GB VRAM, may fail on ZeroGPU)")
+    ],
+    outputs=gr.Video(),
+    title=cfg.TITLE,
+    description=cfg.DESCRIPTION,
+    examples=cfg.EXAMPLES,
+    article=cfg.BUY_ME_A_COFFE
+)
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Center the title and description using custom CSS
+interface.css = """  
+    .interface-title {  
+        text-align: center;  
+    }  
+    .interface-description {  
+        text-align: center;  
+    }  
+"""
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Launch the application
+if __name__ == "__main__":
+    interface.launch()

config.py

@@ -0,0 +1,15 @@
+TITLE = "Mochi 1 - Video Generator 🌟"
+
+DESCRIPTION = """  
+✨ Generate a video based on a text prompt using the Mochi-1-preview.                                             
+"""
+
+MODEL_PRE_TRAINED_ID = "genmo/mochi-1-preview"
+
+EXAMPLES = [
+    ["A majestic dragon soaring over a fiery volcano during a lunar eclipse, casting long shadows on the ancient ruins below. The ground trembles as molten lava flows down the mountainside, and the air is thick with ash and smoke.", 70, 24],
+    ["A bustling steampunk city with towering airships navigating the skies, clockwork robots performing various tasks, and intricate networks of pipes and gears running through every building. The streets are filled with people in Victorian attire, and the sky is a perpetual haze of smoke and steam.", 60, 30],
+    ["A magical forest filled with luminescent creatures that glow softly in the dim light, ancient ruins covered in mystical runes, and a crystal-clear river flowing under the light of a full moon. The trees are tall and twisted, their branches intertwined to form a canopy that filters the moonlight.", 50, 25],
+    ["An underwater city inhabited by mermaids, surrounded by vibrant coral reefs and schools of colorful fish. The buildings are constructed from seashells and coral, with bioluminescent plants lighting the streets. The water is clear and teeming with marine life, creating a serene and otherworldly atmosphere.", 65, 20],
+    ["A futuristic space station orbiting a distant exoplanet, equipped with massive solar panels and docking bays filled with various starships. The observation decks offer breathtaking views of the galaxy, with nearby nebulas and distant stars twinkling in the vast expanse of space.", 75, 30]
+]

requirements.txt

@@ -0,0 +1,7 @@
+accelerate
+torch
+gradio
+transformers
+git+https://github.com/huggingface/diffusers
+sentencepiece
+opencv-python