nada

app.py

@@ -1,5 +1,4 @@
-import spaces  # Import ZeroGPU compatibility module
-import torch
+import spaces
 import imageio
 import os
 import gradio as gr
@@ -15,9 +14,6 @@ from huggingface_hub import snapshot_download
 weights_dir = './allegro_weights'
 os.makedirs(weights_dir, exist_ok=True)
 
-
-# Download the necessary model files if not in shared UI
-
 print(f"Downloading models to {weights_dir}...")
 snapshot_download(
     repo_id='rhymes-ai/Allegro',
@@ -107,9 +103,9 @@ def process_pipeline(user_prompt, guidance_scale, num_sampling_steps, seed, enab
     out_video = allegro_pipeline(
         user_prompt, 
         negative_prompt=negative_prompt, 
-        num_frames=88,
-        height=720,
-        width=1280,
+        num_frames=30,
+        height=360,
+        width=640,
         num_inference_steps=num_sampling_steps,
         guidance_scale=guidance_scale,
         max_sequence_length=512,