Update

app.py

@@ -27,8 +27,6 @@ if torch.cuda.is_available():
     pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
     pipe.enable_model_cpu_offload()
     pipe.enable_vae_slicing()
-else:
-    pipe = None
 
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
@@ -37,7 +35,8 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 
-def to_video(frames: list[np.ndarray], fps: int) -> str:
+def to_video(frames: np.ndarray, fps: int) -> str:
+    frames = np.clip((frames * 255), 0, 255).astype(np.uint8)
     out_file = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
     writer = imageio.get_writer(out_file.name, format="FFMPEG", fps=fps)
     for frame in frames:
@@ -62,7 +61,7 @@ def generate(
         width=576,
         height=320,
         generator=generator,
-    ).frames
+    ).frames[0]
     return to_video(frames, 8)