Upload app.py

app.py

@@ -27,26 +27,19 @@ def process_video(video_path: str):
     T, H, W = video_tensor.shape
 
     temp_path = tempfile.mktemp(suffix=".mp4")
-
-    # Normalize video to [0,1]
     video_tensor = np.clip(video_tensor, 0, 1)
+    cmap = cm.get_cmap("inferno")
 
     writer = imageio.get_writer(
         temp_path, fps=30, codec="libx264", ffmpeg_params=["-pix_fmt", "yuv420p"]
     )
 
-    cmap = cm.get_cmap("inferno")
-    # for t in range(T):
-    #     frame = video_tensor[t]
-    #     colored = (cmap(frame)[..., :3] * 255).astype(np.uint8)
-    #     writer.append_data(colored)
-    # writer.close()
-    with iio.imopen(temp_path, "w", plugin="ffmpeg", fps=30, codec="libx264", pixelformat="yuv420p") as vid:
-        for t in range(T):
-            frame = video_tensor[t]
-            colored = (cmap(frame)[..., :3] * 255).astype(np.uint8)
-            vid.append_data(colored)
+    for t in range(T):
+        frame = video_tensor[t]
+        colored = (cmap(frame)[..., :3] * 255).astype(np.uint8)
+        writer.append_data(colored)
 
+    writer.close()
     return video_path, temp_path
 
 # Unified processing