fix layout

app.py

@@ -1,4 +1,5 @@
 import os
+
 if os.environ.get("SPACES_ZERO_GPU") is not None:
     import spaces
 else:
@@ -7,6 +8,7 @@ else:
         def GPU(func):
             def wrapper(*args, **kwargs):
                 return func(*args, **kwargs)
+
             return wrapper
 
 import torch
@@ -22,33 +24,53 @@ pipe = MochiPipeline.from_pretrained(cfg.MODEL_PRE_TRAINED_ID, variant="bf16", t
 pipe.enable_model_cpu_offload()
 pipe.enable_vae_tiling()
 
+
 @spaces.GPU(duration=600)
-def generate_video(prompt, num_frames=84, fps=30):
-    # Generate video frames
-    print("Generating video frames...")
-    frames = pipe(prompt, num_frames=num_frames).frames[0]
+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
+        # 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.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
-
+    article=cfg.BUY_ME_A_COFFE,
+    layout="vertical"  # Use vertical layout to center title and description
 )
 
+# Center the title and description using custom CSS
+interface.css = """  
+    .interface-title {  
+        text-align: center;  
+    }  
+    .interface-description {  
+        text-align: center;  
+    }  
+"""
+
 # Launch the application
 if __name__ == "__main__":
     interface.launch()

config.py

@@ -1,6 +1,3 @@
-import os
-import numpy as np
-
 TITLE = "Mochi 1 - Video Generator 🌟"
 
 DESCRIPTION = """  
@@ -19,21 +16,11 @@ BUY_ME_A_COFFE = """
 
 MODEL_PRE_TRAINED_ID = "genmo/mochi-1-preview"
 
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
-DEFAULT_SEED = 42
-DEFAULT_RANDOMIZE_SEED = False
-DEFAULT_WIDTH = 1024
-DEFAULT_HEIGHT = 1024
-DEFAULT_GUIDANCE_SCALE = 4.5
-DEFAULT_NUM_INFERENCE_STEPS = 40
-DEFAULT_LORA_SCALE = 4.5
-
 EXAMPLES = [
-     ["A beautiful sunrise over the mountains 🌄", 100, 24],
-    ["A futuristic city with flying cars 🚗✨", 120, 30],
-    ["A serene beach with waves crashing 🌊", 80, 25],
-    ["A bustling market in a medieval town 🏰", 90, 20],
-    ["A spaceship traveling through space 🚀", 110, 30]
+    ["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]
 ]